5.gpu版本Pytorch执行.cuda()代码会卡死

在使用GPU版本的PyTorch时,遇到.model.cuda()操作导致程序卡死的问题。尝试了更换不同版本的PyTorch、PyCharm以及在Python解释器中直接运行,问题依然存在。最终发现并非代码或工具版本问题,而是服务器故障所致,重启服务器后问题得到解决。此外,分享了类似问题的解决建议,如容器内PyTorch的CUDA问题。

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

下载好了gpu版本的pytorch后测试

发现torch.cuda.is_available() = True了,但是.cuda()命令运行时候一直卡死,没有反应:
在这里插入图片描述
使用pycharm远程连接也是这样,一直卡死在model.cuda()语句不动:

在这里插入图片描述
怀疑了好久是不是版本有问题,尝试了各种pytorch-conda-python版本都没有用,而且我下载的就是官网推荐的版本,所以应该是没问题的。后来甚至还怀疑是不是pycharm的问题,但是上面也试了脱离pycharm直接在python虚拟环境里面的python解释器直接运行.cuda()也不行,所以也不是pycharm的问题。
附各种版本尝试,呜呜呜
在这里插入图片描述

就一直觉得问题很奇怪,最后还是请教同学跑了一下他那里的代码想看看到底怎么使用GPU的,发现他那里也不能用了,结果是因为服务器有问题了,重启了一下服务器就万事大吉了!不得不感叹重启大法确实好,我一直在找自己这里的问题,卡了很多天,反正

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值