ubuntu 18.04安装NVIDIA-docker

如果你直接用百毒搜,得到的都是错误答案,因为官网更新了,哈哈
百毒,csdn,祸害互联网,祸害IT界

官网链接:
https://nvidia.github.io/nvidia-docker/

正确示范

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
  sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update

安装完后,就是需要看看配置文件:

$ sudo cat /etc/docker/daemon.json
{
    "registry-mirrors": ["http://f1361db2.m.daocloud.io"],
    "runtimes": {
        "nvidia": {
            "path": "nvidia-container-runtime",
            "runtimeArgs": []
        }
    }
}

重启一下服务

sudo service docker restart

跑一下demo

docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi
发布了281 篇原创文章 · 获赞 35 · 访问量 6万+

猜你喜欢

转载自blog.csdn.net/TQCAI666/article/details/102567072