Ubuntu 20.04 サーバーにcuda11.2とcudnn8.1をインストールする

新しく導入したRTX3090を積んだUbuntu 20.04 サーバーでTensorflow2を利用するために、cuda11.2とcudnn8.1をインストールした際の手順をまとめた記事になります。

状況

執筆時点での最新版であるTensorflow-2.5を利用したい。
以下のサイトでテスト済のCUDA, cuDNNを確認したところそれぞれ11.2, 8.1であった。
https://www.tensorflow.org/install/source#linux

必要なパッケージのダウンロード

CUDA

以下からダウンロード
https://developer.nvidia.com/cuda-11.2.2-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=2004&target_type=deblocal

cuDNN

cuDNNを利用するにはNVIDIA Developerアカウントを作成する必要があります。
ログインできるようになると以下のサイトからインストールしたいパッケージを選択してダウンロードしてください。
https://developer.nvidia.com/rdp/cudnn-archive

今回は以下の名前のパッケージ２つをダウンロードしています。
これは適宜ご自身のサーバーに合わせてください（基本的には以下の２つのケースがほとんどかと思います）。

• cuDNN Runtime Library for Ubuntu20.04 x86_64 (Deb)
• cuDNN Developer Library for Ubuntu20.04 x86_64 (Deb)

インストール方法は以下を参照
https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#installlinux-deb

sudo dpkg -i libcudnn8_x.x.x-1+cudax.x_amd64.deb
sudo dpkg -i libcudnn8-dev_8.x.x.x-1+cudax.x_amd64.deb

インストール

CUDA

公式の手順に則ってやるだけ。

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/11.2.2/local_installers/cuda-repo-ubuntu2004-11-2-local_11.2.2-460.32.03-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-2-local_11.2.2-460.32.03-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-2-local/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

インストール中のメッセージに以下のようなものが出てると思います。

A modprobe blacklist file has been created at /etc/modprobe.d to prevent Nouveau
from loading. This can be reverted by deleting the following file:
/etc/modprobe.d/nvidia-graphics-drivers.conf

A new initrd image has also been created. To revert, please regenerate your
initrd by running the following command after deleting the modprobe.d file:
`/usr/sbin/initramfs -u`

*****************************************************************************
*** Reboot your computer and verify that the NVIDIA graphics driver can   ***
*** be loaded.                                                            ***
*****************************************************************************

色々なサイトは手動でNouveauを無効化していますが、設定ファイルを自動で作成してくれているので再起動するだけで十分です。

おわりに

CUDAの設定は結構頻繁にやっていて毎回調べながらやるのも大変だったので備忘録としてまとめました。
ただしインストール方法は変わっていくと思われますので、都度公式ドキュメントは確認することをお勧めします。

それでは。

参照

• テスト済みのビルド構成
• Debian Local Installation