cuda error

device-side assert triggered

Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.

可能引起的原因：

• 数据集太大，GPU显存不足

• 分类任务标签个数对应不上

• loss function定义错误

可以尝试以下方法解决：

1、优化深度神经网络的性能，通过下面两个设置

import torch
torch.backends.cudnn.enable = True
torch.backends.cudnn.benchmark = True

2、检查分类任务标签数量

3、换成cpu运行，就会显示完整的错误信息了

参考链接：关于cuda error:device-side assert triggered的3种解决办法 - 知乎 (zhihu.com)