在使用GPU版本的PyTorch时，遇到.model.cuda()操作导致程序卡死的问题。尝试了更换不同版本的PyTorch、PyCharm以及在Python解释器中直接运行，问题依然存在。最终发现并非代码或工具版本问题，而是服务器故障所致，重启服务器后问题得到解决。此外，分享了类似问题的解决建议，如容器内PyTorch的CUDA问题。
下载好了gpu版本的pytorch后测试
发现torch.cuda.is_available() = True了，但是.cuda()命令运行时候一直卡死，没有反应：
使用pycharm远程连接也是这样，一直卡死在model.cuda()语句不动：
怀疑了好久是不是版本有问题，尝试了各种pytorch-conda-python版本都没有用，而且我下载的就是官网推荐的版本，所以应该是没问题的。后来甚至还怀疑是不是pycharm的问题，但是上面也试了脱离pycharm直接在python虚拟环境里面的python解释器直接运行.cuda()也不行，所以也不是pycharm的问题。
附各种版本尝试，呜呜呜
就一直觉得问题很奇怪，最后还是请教同学跑了一下他那里的代码想看看到底怎么使用GPU的，发现他那里也不能用了，结果是因为服务器有问题了，重启了一下服务器就万事大吉了！不得不感叹重启大法确实好，我一直在找自己这里的问题，卡了很多天，反正
最低0.47元/天 解锁文章
扫一扫
3118
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
