Ubuntu18.04にNVIDIADocker2をインストールします[利用可能な20210225を確認します]

 

 

 

背景：NVIDIA転送をインストールするには、環境によって異なります。DockerでGPUをサポートするために、NVidiaはnvidia-docker2を取得していましたが、現在はNVIDIA ContainerToolkitにアップグレードされています。

githubを参照してください

---

インストール

1. Dockerインストールポータル

上記のDockerのインストールが完了したら、nvidia-dockerソースの構成を開始します

1. ソースを追加

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. nvidia-container-toolkitをインストールし、dockerを再起動します

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

•  

テスト

1. 最新のCUDA公式マップを使用してnvidia-smiをテストします

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

2.デュアルGPUをサポートするコンテナーを起動します

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

1. GPU 1を指定し、コンテナーを実行します

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

1. パーソナルコンテナのパフォーマンスを指定する（めったに使用されない）

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

参照：https：//blog.csdn.net/weixin_38369492/article/details/105809571