容器里 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

这样就解决了。

  • 作者:yimingfei
  • 原文链接:https://blog.csdn.net/yimingfei/article/details/110184611
    更新时间:2022-10-13 12:15:43