ubuntu20.04安装NVIDIA-docker_ubuntu20安装nvidia-docker

安装前提：

1、显卡驱动安装。先确保你已经安装了NVIDIA的显卡驱动。你可以通过运行nvidia-smi命令来检查驱动是否已经安装。如果驱动已经安装，这个命令会显示你的GPU的详细信息。

2、安装docker。确保docker已经正确安装。

安装NVIDIA-docker

你可以运行以下命令来安装：

# 添加NVIDIA的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
sudo apt-get update
sudo apt-get install -y nvidia-docker2

1
2
3
4
5
6
7
8

重新启动docker:

sudo systemctl restart docker
1

现在，你应该已经成功安装了NVIDIA-docker，你可以通过运行docker run --gpus all nvidia/cuda:10.0-base nvidia-smi来测试是否可以在Docker容器中使用你的GPU。