在我用服务器跑神经网络时,出现了这个问题,反复在网上查找解决方案,感觉网络上的方案无法解决问题,于是开始自己尝试解决。
实际原因
出现这个问题的实际原因应该是,显卡与pytorch版本不一致,导致无法正常调用cuda进行数据处理
在我这个服务器上,RTX2080并没有更新驱动以适配pytorch.1.7.1,然而我的pytorch是1.8.0
解决方案
更改pytorch为1.2.0,问题解决
版权声明:本文为zhou_zechu原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。