解决： RuntimeError: CUDA error: device-side assert triggered CUDA kernel errors might be

最新推荐文章于 2025-04-17 11:04:18 发布

我想吃鱼了，

最新推荐文章于 2025-04-17 11:04:18 发布

阅读量4.4k

点赞数

分类专栏： bug调试文章标签： python 深度学习开发语言

本文链接：https://blog.youkuaiyun.com/python_innocent/article/details/129803704

版权

bug调试专栏收录该内容

9 篇文章

订阅专栏

文章讨论了在使用PyTorch构建神经网络时可能遇到的问题，包括传入传出通道数不匹配、网络最终全连接层输出类别与标签不符、设备指定不明确以及设备管理问题。作者建议通过创建单个tensor来测试网络的输入输出正确性，并提供了一段示例代码进行排查。在尝试修改和关闭再打开后，问题得到解决。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

主要可能出现的原因：1.传入传出的通道数不对；2.net最终全连接输出的class数目对不上labels；3. devices没有具体到某一个，而是一个list;4. 关了重新再连。建议先用一个单tensor如下，进行检验net的输入输出正确性，再放到dataset上。

X = torch.rand(size=(32, 3, 224, 224)).to(devices[0])
for layer in net:
    X = layer(X)
    print(layer.__class__.__name__,'output shape:\t', X.shape)

笔者首先尝试了前三个，结果还是报这个错，确认修改无误了把这玩意关了，再打开，正常运行了。。。