Install NVIDIA Docker 2 on Ubuntu 18.04 [verify available 20210225]

 

 

 

Background: To install NVIDIA transfer, it depends on the environment. In order to support GPU in docker, NVidia used to get a nvidia-docker2, now it has been upgraded to NVIDIA Container Toolkit.

Refer to github

---

installation

1. Docker installation portal

After completing the above Docker installation, start to configure the nvidia-docker source

1. Add source

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

•  

3. Install nvidia-container-toolkit and restart docker

sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker

•  

test

1. Use the latest CUDA official map to test nvidia-smi

$ sudo docker run --gpus all nvidia/cuda:10.0-base nvidia-smi

2. Launch a container that supports dual GPUs

$ sudo docker run --gpus 2 nvidia/cuda:10.0-base nvidia-smi

1. Specify GPU 1, run the container

$ sudo docker run --gpus 2 nvidia/cuda:9.0-base nvidia-smi

1. Specify personal container performance (rarely used)

$ sudo docker run --gpus all,capabilities=utility nvidia/cuda:10.0-base nvidia-smi

Reference: https://blog.csdn.net/weixin_38369492/article/details/105809571