Ubuntu安装Nvidia-Docker

安装前提：

（1）已经在宿主机安装好了docker，可执行docker -v命令验证；

（2）已经在宿主机上安装好了nvidia驱动，可执行nvidia-smi验证。

安装Nvidia-Docker:

# 添加Nvidia-Docker的GPG key
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -

# 添加Nvidia-Docker的repository
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

# 更新apt-get
sudo apt-get update

# 安装Nvidia-Docker
# 方法一：
# 安装Nvidia-Docker, Nvidia-Docker有两个版本，版本2为新版，此处安装的为版本2
sudo apt-get install -y nvidia-docker2
/
# 方法二：
# 或者直接安装nvidia-container-toolkit，nvidia-container-toolkit是一组软件包,
# 它们包括一个容器运行时库和一些工具，可以自动配置容器来利用NVIDIA GPU。
# 它支持不同的容器引擎，如Docker、containerd、LXC、Podman等。(推荐)
sudo apt-get install -y nvidia-container-toolkit

# docker服务重新启动
sudo systemctl restart docker

测试（有nvidia-smi显卡信息输出表示安装成功）：

sudo docker run --rm --gpus all nvidia/cuda:${根据nvidia-smi & docker-hub nvidia官方查询得到} nvidia-smi

# 例如 cuda:11.7.1-base-ubuntu20.04
sudo docker run --rm --gpus all nvidia/cuda:11.7.1-base-ubuntu20.04 nvidia-smi