UbuntuにCUDAをインストール及びDocker環境の構築

環境：Ubuntu 20.04.5 LTS

下記のコマンドを実行

step1：下記のコマンドを実行し、Cudaをインストール

参考サイト：https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=20.04&target_type=deb_local

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda-repo-ubuntu2004-12-0-local_12.0.0-525.60.13-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-12-0-local_12.0.0-525.60.13-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-12-0-local/cuda-5E22DB91-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda
reboot

step２：下記のコマンドを実行し、cuda-toolkitとcuda-driverをインストール

sudo apt install nvidia-cuda-toolkit nvidia-driver-525
reboot

step３：下記のコマンドを実行し、インストールが成功したのかをチェック

nvidia-smi

step4: nvidia-dockerをインストール

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

step5: 下記のコマンドを実行し、nvidia-docker環境構築が成功したのかをチェック

docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi