在容器里,模型运行到model.cuda()这一步很久不动,直到被kill。

解决方案:

查看容器里cuda为10.2版本,但是torch安装的是0.4.0版本,所以我重新安装为匹配的版本:

pip3 install torch==1.6.0 torchvision==0.7.0

这样就解决了。

Logo

权威|前沿|技术|干货|国内首个API全生命周期开发者社区

更多推荐