1.网络安装
# Add the package repositories
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 && sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker
2.离线安装
Index of /nvidia-docker/libnvidia-container/stable/ubuntu16.04/amd64/https://mirror.cs.uchicago.edu/nvidia-docker/libnvidia-container/stable/ubuntu16.04/amd64/在上面的网站上,下载以下文件:
libnvidia-container1_1.9.0-1_amd64.deb
libnvidia-container-tools_1.9.0-1_amd64.deb
nvidia-container-runtime_3.9.0-1_all.deb
nvidia-container-toolkit_1.9.0-1_amd64.deb
nvidia-docker2_2.9.1-1_all.deb
上传到服务期后,使用sudo dpkg -i *.deb进行安装
3.验证
docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi
如果成功显示GPU信息,则说明安装成功