本記事の目的

Ubuntu環境におけるNvidiaドライバおよびCUDA Toolkitのインストール

Ubuntu環境におけるNVIDIAドライバおよびCUDAのインストール方法について、なかなかうまくいかなかったので、うまくいった方法を紹介します。

環境

        `:+ssssssssssssssssss+:`           --------------------- 
      -+ssssssssssssssssssyyssss+-         OS: Ubuntu 24.04 LTS x86_64 
    .ossssssssssssssssssdMMMNysssso.       Host: A620I AX 
   /ssssssssssshdmmNNmmyNMMMMhssssss/      Kernel: 6.8.0-39-generic 
  +ssssssssshmydMMMMMMMNddddyssssssss+     Uptime: 3 hours, 38 mins 
 /sssssssshNMMMyhhyyyyhmNMMMNhssssssss/    Packages: 2214 (dpkg), 17 (snap) 
.ssssssssdMMMNhsssssssssshNMMMdssssssss.   Shell: bash 5.2.21 
+sssshhhyNMMNyssssssssssssyNMMMysssssss+   Resolution: 1920x1080, 1920x1080 
ossyNMMMNyMMhsssssssssssssshmmmhssssssso   DE: GNOME 46.0 
ossyNMMMNyMMhsssssssssssssshmmmhssssssso   WM: Mutter 
+sssshhhyNMMNyssssssssssssyNMMMysssssss+   WM Theme: Adwaita 
.ssssssssdMMMNhsssssssssshNMMMdssssssss.   Theme: Yaru-bark-dark [GTK2/3] 
 /sssssssshNMMMyhhyyyyhdNMMMNhssssssss/    Icons: Yaru-bark [GTK2/3] 
  +sssssssssdmydMMMMMMMMddddyssssssss+     Terminal: gnome-terminal 
   /ssssssssssshdmNNNNmyNMMMMhssssss/      CPU: AMD Ryzen 7 7700X (16) @ 5.573G 
    .ossssssssssssssssssdMMMNysssso.       GPU: NVIDIA GeForce RTX 4060 Ti 16GB 
      -+sssssssssssssssssyyyssss+-         GPU: AMD ATI 0f:00.0 Raphael 
        `:+ssssssssssssssssss+:`           Memory: 12125MiB / 31196MiB 
            .-/+oossssoo+/-.

方法

【準備】　Ubuntu残る、既存ドライバの削除

すでにNvidiaドライバやCUDA Toolkitがインストールされており、バージョンアップ（ダウン）を行う場合にはこの手順を踏む必要があります。

① ドライバの削除

下記のコマンドをターミナルで入力し、残存するドライバ等の削除を行います。

$ sudo apt --purge remove -y nvidia-*
$ sudo apt --purge remove -y cuda-*
$ sudo apt --purge remove -y libcudnn*
$ sudo apt --purge remove -y cudnn-*
$ sudo apt autoremove -y

完了後、再起動を行います。
再起動後、互換ドライバがあたっているかと思います。

1. NVIDIAドライバのインストール

ここからは、ドライバのインストールを行っていきます。

① リポジトリの追加

CUDA Toolkit Archive( https://developer.nvidia.com/cuda-toolkit-archive )にアクセスします。

インストールしたいバージョンを選択します。

今回は、執筆時点で最新版のCUDA 12.6のインストールを想定しています。

Operating System, Architecture, Distribution, Installer Typeを以下のように選択

Versionは、自身の環境に合わせて適当に選んでください。

このようにすると、Base InstallerとDriver Installerのインストールコマンドが下段にそれぞれ表示されると思います。基本的には、Base Installerの指示に沿って行っていくのですが、
3行目までを実行した後、CUDA Toolkitのインストールをそのまま行わず、Nvidiaドライバのインストールを行います。

おそらく、イメージがつかないかと思いますので例を挙げて説明していきます。

例：CUDA Toolkit 12.6の場合

サイトに記載のBase Installerのインストール方法は以下のように記述されています。

Base Installer.bash

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb
$ sudo dpkg -i cuda-keyring_1.1-1_all.deb
$ sudo apt-get update
$ sudo apt-get -y install cuda-toolkit-12-6

このとき、3行目のsudo apt-get updateまでを実行し、**4行目のsudo apt-get -y install cuda-toolkit-12-6は実行せずに、次のステップに進みます。

② NVIDIAドライバのインストール

先程のステップまでで、ドライバのリポジトリ登録は完了しましたので、続いてドライバのインストールを行います。

Ubuntuのプリインストールアプリの追加のドライバーを開きます。

ここで、インストールしたいバージョンのNvidiaドライバを選択し、変更の適用を押して、インストールを行ってください。

インストールが完了すると再起動を促されますので、 （もし促されなかった場合でも） 再起動してください。

インストールするドライバのバージョン
ドライバのバージョンは、基本的にお好きなバージョンを入れれば良いのですが、
CUDA Toolkitで推奨するバージョンをインストールしてください

2. CUDA Toolkitのインストール

ここまできたら、CUDA Toolkitをインストールしていきます。

再度、先程のサイトのBase Installerの欄を確認してください。
先程までのステップで、3行目まで実行したはずですので、続いて4行目のコマンドを入力してください。

Base Installer.bash

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb
$ sudo dpkg -i cuda-keyring_1.1-1_all.deb
$ sudo apt-get update
$ sudo apt-get -y install cuda-toolkit-12-6　← これを実行

すると、CUDA Toolkitのインストールが始まります。環境によって待ち時間はまちまちですが、私の環境では10分程度かかりましたので、気長に待ちましょう。

3. 環境変数の追加

最後に環境変数の追加を行っていきます。

バージョンの変更等で、環境を変更した場合は環境変数の追加は不要な場合があります。
後述の動作確認を行ってみてうまくいかないようであれば、この手順を実施すると良いでしょう。

~/.bashrcを管理者権限を付与したテキストエディタ等で開いてください。
ここでは、Ubuntuのデフォルトシェルであるbashを例にしていますが、zshなどの別のシェルをお使いの場合は、そのシェルに対応したrcファイルを開いてください。

開けましたら、次の記述を末尾に追加してください。

~/.bashrc

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

追加できたら、保存を行ってシェルの設定を行います。

$ source ~/.bashrc

何もエラーが表示されなければ、ここまででインストール作業は終了です。
お疲れ様でした！

動作確認

最後に動作確認を行います。ターミナルで以下のコマンドを入力してください。

$ nvidia-smi

user@hoge:~$ nvidia-smi

Mon Aug  5 12:17:00 2024       
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 555.42.06              Driver Version: 555.42.06      CUDA Version: 12.5     |
|-----------------------------------------+------------------------+----------------------+
| 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 4060 Ti     Off |   00000000:01:00.0  On |                  N/A |
|  0%   56C    P2             34W /  165W |    1392MiB /  16380MiB |      5%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+
                                                                                         
+-----------------------------------------------------------------------------------------+
| Processes:                                                                              |
|  GPU   GI   CI        PID   Type   Process name                              GPU Memory |
|        ID   ID                                                               Usage      |
|=========================================================================================|
|    0   N/A  N/A      2841      G   /usr/lib/xorg/Xorg                            228MiB |
|    0   N/A  N/A      3086      C   ...libexec/gnome-remote-desktop-daemon        120MiB |
|    0   N/A  N/A      3149      G   /usr/bin/gnome-shell                           57MiB |
|    0   N/A  N/A      3236      G   ...libexec/gnome-remote-desktop-daemon          0MiB |
|    0   N/A  N/A      3648      G   /usr/libexec/xdg-desktop-portal-gnome          14MiB |
|    0   N/A  N/A      3991      G   ...seed-version=20240802-130108.496000         96MiB |
|    0   N/A  N/A      5665      G   /usr/bin/gnome-control-center                  17MiB |
|    0   N/A  N/A      6815      G   ...erProcess --variations-seed-version         73MiB |
|    0   N/A  N/A      8055      G   ...nglingPtr --variations-seed-version         57MiB |
|    0   N/A  N/A      8536      G   ...yOnDemand --variations-seed-version         27MiB |
|    0   N/A  N/A     36891      G   /usr/bin/nautilus                              14MiB |
|    0   N/A  N/A     37979      C   .../.anaconda3/envs/PyTorch/bin/python        618MiB |
+-----------------------------------------------------------------------------------------+

細かい表示は異なりますが、このような画面が表示されるはずです。
このようになっていれば、NVIDIAドライバは正しくインストールされています。

続いて、以下のコマンドを入力してください。

$ nvcc -V

すると、

user@hoge:~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Thu_Jun__6_02:18:23_PDT_2024
Cuda compilation tools, release 12.5, V12.5.82
Build cuda_12.5.r12.5/compiler.34385749_0

となれば正しくCUDA Toolkitがインストールされています。
エラーとなった場合は、~/.bashrcを確認してみましょう。

参考文献