1,171
社区成员
发帖
与我相关
我的任务
分享
我尝试使用一个cuda版本的pytorch的docker 镜像。
我在windows10上拉取该镜像创建容器后,nvidia-smi命令有效,并且训练时cuda可用。于是我使用docker save xx.tar xx导出该容器后,在ubuntu上通过docker load < xx.tar导入镜像,发现nvidia-smi命令不可用了。请问这可能是怎么回事呢?
你试试在save的时候,最后加上 REPOSITORY:TAG 的信息,看看能否解决问题