526互联

Ubuntu 部署 nvidia-docker

发布时间 2023-07-25 23:23:48作者: BonnieBlueButler

1）安装nvidia显卡驱动

2）安装 Docker
参考：Docker清华源 ,
Daocloud Docker 极速下载
$ export DOWNLOAD_URL="https://mirrors.tuna.tsinghua.edu.cn/docker-ce"
$ curl -sSL https://get.daocloud.io/docker | sh

修改 Docker 镜像服务器
{ "runtimes": { "nvidia": { "path": "nvidia-container-runtime", "registry-mirrors": [ "https://registry.docker-cn.com", "http://hub-mirror.c.163.com", "https://docker.mirrors.ustc.edu.cn" ], "runtimeArgs": [] } } }

重启 Docker 服务:
$ sudo systemctl daemon-reload
$ sudo systemctl restart docker

3）安装nvidia-docker软件包
参考：Docker容器如何优雅使用NVIDIA GPU ,
nvidia-docker github ,
docker cuda版本
``
将工具包的包存储库添加到您的系统
$ distribution=$(. /etc/os-release;echo $ID$VERSION_ID) && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - && curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

安装nvidia-docker2软件包
$ sudo apt-get update
$ sudo apt-get install -y nvidia-docker2
重启 Docker 守护进程
$ sudo systemctl restart docker