解决方法：
在代码前加上torch.backends.cudnn.enabled = False。
参考：https://blog.csdn/qq_33866063/article/details/121009069
解决这个问题后又报 CUDA error: device-side assert triggered
经过调式发现是Unable to get repr for <class 'torch.Tensor'> 错误信息定位在损失函数处。
出错原因：使用BCE损失函数的tensor没有进行（0-1）之间的映射。将input进行sigmoid处理即可。
参考：unable to get repr for ＜class ‘torch.Tensor‘＞错误可能原因_peacefairy的博客-CSDN博客
更多推荐
pytorch bug: Unable to find a valid cuDNN algorithm to run convolution
发布评论