警告如下:
首先python里面警告一般不用管这个也一样,但是一定要消除这个警告需要如下解决方法:
方法一:
在代码中加入以下代码:
torch.backends.cudnn.enabled=False
运行
禁用cudnn计算,报错消失,但是运行速度会变得很慢。
方法二:
减小batchsize大小。
方法三(推荐):
查看自己的pytorch 版本是不是2.3.0, 在 pytorch 2.3.0 版本中,即使没有抛出异常,它也会打印这个不需要的警告:可以参阅https://github.com/pytorch/pytorch/pull/125790
如果你想摆脱这个警告,可以恢复到 torch 2.2.2(然后你还必须将 torchvision 恢复到 0.17.2):
pip3 install torchvision==0.17.2
pip3 install torch==2.2.2
如果嫌弃版本太低将torch版本升级到>=2.3.1即可消除警告