Ubuntu 20.04 に NVIDIA の CUDA のインストール方法です。私は主に Google Cloud や Amazon EC2 の GPU インスタンスを使っているので、それ前提で書きます。ストレージは20GB程度必要です。Python からしか使わない場合は Anconda 内蔵の CUDA を使用するのがお勧めです。

Ubuntu の他のバージョンはこちらをご覧ください。

• Ubuntu 16.04 https://qiita.com/yukoba/items/3692f1cb677b2383c983
• Ubuntu 18.04 https://qiita.com/yukoba/items/4733e8602fa4acabcc35

前準備

sudo apt update
sudo apt upgrade

CUDA 10や11をNVIDIAのリポジトリからインストール

Ubuntu 18.04 向けが https://developer.nvidia.com/cuda-downloads で配布されているのですが、Ubuntu 20.04 はまだ配布されていないのですが、Ubuntu 18.04 用がそのまま動きます。以下は、Installer Type で deb(network) を選択する場合の流れです。ここで apt install で cuda だと最新版がインストールされ、cuda-10-1 のようにバージョン指定も出来ます。

sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub

wget wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.2.89-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804_10.2.89-1_amd64.deb
sudo apt update

sudo apt install cuda cuda-drivers
sudo reboot

rm cuda-repo-ubuntu1804_10.2.89-1_amd64.deb

その後、~/.bashrc の末尾にこれを追加する必要があります。

export PATH="/usr/local/cuda/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"

そして一度ログオフして再度ログインしてください。

nvidia-smi で動作確認です。cuBLAS など cuDNN 以外は一通りついてきます。

cuDNNのインストール

cuDNN は別配布ですが、こちらも Ubuntu 18.04 用がインストールできます。

echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" | sudo tee /etc/apt/sources.list.d/nvidia-ml.list
sudo apt update
sudo apt install libcudnn7-dev

CUDA 10.1をUbuntuのリポジトリからインストール

NVIDIA のリポジトリからではなく、Ubuntu 20.04 のリポジトリからインストールする方法です。

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

nvidia-smi で動作確認です。cuBLAS など cuDNN 以外は一通りついてきます。

ただし、この方法は Ubuntu 20.04 である限り、ずっと CUDA 10.1 です。

Anaconda内蔵のCUDAとcuDNNを使用する場合

Anaconda で conda install tensorflow-gpu などをすると自動的に CUDA と cuDNN がインストールされます。ただしデバイスドライバはインストールされないので sudo apt install nvidia-driver-440 して sudo reboot が必要です。Python からしか使わない場合は、この方法が簡単です。NVIDIA のリポジトリは不要です。

GNOME Display Manager を止める

NVIDIA のリポジトリからのインストールでも、Ubuntu 20.04 のリポジトリからのインストールでも、ディスプレイマネージャーの GNOME Display Manager が動いてしまいます。普通のパソコンでは動くべきですが、クラウドで動かしている場合は（大きな害はないですが）無意味です。

以下のコマンドで止まります。

sudo systemctl set-default multi-user
sudo reboot

もし復活させたい場合は、以下のコマンドで復活します。

sudo systemctl set-default graphical
sudo reboot

後始末

sudo apt autoremove
sudo apt clean