Ubuntu22.04 LTSにCUDA Toolkitをインストールする

タイトル通りですが、Ubuntu22.04にCUDAをインストールします。

環境

OS: Ubuntu 22.04
CPU: AMD Ryzen 5 2600
GPU: Nvidia RTX2070

CUDA Toolkitの対応バージョンを確認

下記公式サイトで、自分が利用しているNVIDIAのGPUが対応しているCUDA Toolkitのバージョンを確認します。

https://docs.nvidia.com/deeplearning/cudnn/support-matrix/index.html

ほとんどのGPUは、最新（2023/04/10時点）のCUDA Toolkit 12.0系に対応しているはずです。

CUDA Toolkitをダウンロード

公式サイトの指示に従って、Toolkitをダウンロードします。

https://developer.nvidia.com/cuda-downloads

上記サイトの最後に選択する「Installer Type」によってコマンドが異なります。

Toolkitをインストールするパソコンが、どういう環境にあるかで選択すべきものが変わりますが、ここでは個人用のデスクトップパソコンに入れることを想定して、deb(network)を選択したときのコマンドを掲載しておきます。

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb
sudo dpkg -i cuda-keyring_1.0-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda

cudaをインストールしたら、再起動しましょう。

動作確認

端末を開いて、nvidia-smiコマンドを実行しましょう。

利用しているGPUの型番が正しく表示されれば、正常にドライバーはインストールされています。

$ nvidia-smi 
Tue Apr 11 14:24:35 2023       
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 530.30.02              Driver Version: 530.30.02    CUDA Version: 12.1     |
|-----------------------------------------+----------------------+----------------------+
| GPU  Name                  Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf            Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                                         |                      |               MIG M. |
|=========================================+======================+======================|
|   0  NVIDIA GeForce RTX 2070         On | 00000000:06:00.0 Off |                  N/A |
|  0%   50C    P5               11W / 215W|   1890MiB /  8192MiB |     40%      Default |
|                                         |                      |                  N/A |
+-----------------------------------------+----------------------+----------------------+
                                                                                         
+---------------------------------------------------------------------------------------+
| Processes:                                                                            |
|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |
|        ID   ID                                                             Usage      |
|=======================================================================================|
|    0   N/A  N/A      1989      G   /usr/lib/xorg/Xorg                          820MiB |
|    0   N/A  N/A      2120      G   /usr/bin/gnome-shell                        261MiB |
|    0   N/A  N/A      2829      G   ...0287735,18416052544535563571,131072      707MiB |
|    0   N/A  N/A      3060      G   gnome-control-center                         34MiB |
+---------------------------------------------------------------------------------------+