[docker][原创]docker拉取的nvidia/cuda或者带有gpu的镜像无法使用nvidia-smi解决方法

最近玩docker发现自己装的docker容器不能用nvidia-smi,弄了好久终于解决问题。我已经装了docker和nvidia-docker2但是为什么容器里面用不了?

第一确保/etc/docker/daemon.json有如下内容

{

    "runtimes": {

        "nvidia": {

            "path": "/usr/bin/nvidia-container-runtime",

            "runtimeArgs": []

        }

    }

}

没有sudo vi /etc/docker/daemon.json添加一下,我是用了阿里云加速脚本结果把上面内容搞没了。然后

sudo systemctl daemon-reload

sudo systemctl restart docker

最后重新:sudo nvidia-docker run -it IMAGE_ID bash进去即可,注意先前有容器的需要删除掉,发现进去了还是不行,需要重新开个容器

你可能感兴趣的:(docker,深度学习,tensorflow,pytorch,caffe)