nvidia-docker2 github地址:https://github.com/NVIDIA/nvidia-docker

        安装nvidia-docker前要先安装docker。参考:https://blog.csdn.net/weixin_34910922/article/details/125175026

1、移除已安装的nvidia-docker1.0

docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
sudo apt-get purge -y nvidia-docker

2、添加源

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

sudo apt-get update --fix-missing

sudo apt-get install -y nvidia-container-toolkit

sudo apt-get --only-upgrade install nvidia-docker2
sudo systemctl restart docker

3、安装nvidia-docker2并重新加载daemon.json

        选择用新的daemon覆盖之前的。

# On debian based distributions: Ubuntu / Debian
sudo apt-get update

sudo apt-get --only-upgrade install docker-ce nvidia-docker2

sudo systemctl restart docker
# or
sudo pkill -SIGHUP dockerd

4、测试是否安装成功

docker run --runtime=nvidia --rm nvidia/cuda:10.2-devel nvidia-smi

更多推荐