容器里 pytorch model.cuda()卡死 2022-10-13 12:15:43 在容器里,模型运行到model.cuda()这一步很久不动,直到被kill。解决方案:查看容器里cuda为10.2版本,但是torch安装的是0.4.0版本,所以我重新安装为匹配的版本:pip3 install torch==1.6.0 torchvision==0.7.0这样就解决了。