Docker笔记--Ubuntu20.04安装Nvidia-Docker

目录

1--添加软件包存储库

2--安装Nvidia-Docker2

3--测试

---

1--添加软件包存储库

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -

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

sudo apt-get update

2--安装Nvidia-Docker2

        安装 nvidia-docker2 软件包并重新加载 docker 守护程序配置：

sudo apt-get install -y nvidia-docker2

sudo pkill -SIGHUP dockerd

        执行 sudo apt-get install -y nvidia-docker2，可能会出现以下错误：E: Unable to locate package nvidia-docker2，解决方法是完整执行完第 1 步；

3--测试

        重启 docker：

sudo systemctl daemon-reload
sudo systemctl restart docker

        查看显卡信息（执行这步时会自动下载对应的镜像文件）：

docker run --rm --gpus all nvidia/cuda:11.6.2-base-ubuntu20.04 nvidia-smi