NVIDIA コンテナ・ツールキットは,NVIDIA CUDA をコンテナで動かすときに便利.
参考Webページ https://docs.nvidia.com/deploy/cuda-compatibility/index.html
CUDA 対応の GPU であるかを確認のため, 次のコマンドを実行.
lspci | grep -i nvidia | grep VGA
実行結果の例.(表示が空になるときは,NVIDIAのグラフィックス・カードが無い可能性がある).
glxinfo | grep direct
xdriinfo
NVIDIA グラフィックス・カードがある場合に限り,ドライバのインストールを行う
sudo ubuntu-drivers autoinstall
sudo shutdown -r now
システムの再起動後,端末で,次のコマンドを実行し,ドライバ等を確認する
先頭の「Driver Version」のところに,ドライバのバージョンが表示される.NVIDIA CUDA がインストール済みのときは,CUDA のバージョンも表示される.
nvidia-smi
https://github.com/NVIDIA/nvidia-dockerに記載の手順に従う
sudo apt -y update sudo apt -y install containerd docker.io docker-ce docker-ce-cli
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
※ エラーが出る場合には,次のようなコマンドを試してみる.「18.04」のところは Ubuntu のバージョンにあわせて読み替えること.
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu18.04/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt -y update sudo apt -y install nvidia-container-toolkit sudo systemctl restart docker
docker run -it --rm --gpus all nvidia/cuda:10.1-base /bin/bash nvidia-smi exit
docker run -it --rm --gpus all nvidia/cuda /bin/bash nvidia-smi exit