在容器里,模型运行到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
这样就解决了。