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