pytorch错误RuntimeError: CUDA error: device-side assert triggered

在调试模型时出现RuntimeError: CUDA error: device-side assert triggered的错误,经过查找资料,发现是我的label数据是1-37的,在把label数据调整为0-36后,不再报错.
问题主要是标签数据要从0开始,否则就会报出RuntimeError: CUDA error: device-side assert triggered的错误

猜你喜欢

转载自blog.csdn.net/weixin_41963310/article/details/109160405