容器里 pytorch model.cuda()卡死
在容器里,模型运行到model.cuda()这一步很久不动,直到被kill。解决方案:查看容器里cuda为10.2版本,但是torch安装的是0.4.0版本,所以我重新安装为匹配的版本:pip3 install torch==1.6.0 torchvision==0.7.0这样就解决了。...
·
在容器里,模型运行到model.cuda()这一步很久不动,直到被kill。
解决方案:
查看容器里cuda为10.2版本,但是torch安装的是0.4.0版本,所以我重新安装为匹配的版本:
pip3 install torch==1.6.0 torchvision==0.7.0
这样就解决了。
更多推荐
已为社区贡献1条内容
所有评论(0)