毎度、ググっても出てこない小ネタを取り扱っております。
本記事は個人的な見解であり、筆者の所属するいかなる団体にも関係ございません。
0. はじめに
最近、ChatGPTによりLLMとか大流行りですが、サーバーでGPUを使うという事も随分と一般的になってきました。UbuntuサーバーにNVIDIAドライバーを入れる方法も沢山記事があったりしますが、どれも微妙に違っていて何が最適なのかよく分からなかったので、NVIDIA社のオススメの方法を使ってインストールしてみたという記事になります。
あくまでもUbuntuをサーバーとして利用してGPU(RTX)をサーバーとして利用する場合のインストール方法になります。なので、いきなりcudaとかインストールしてしまいます。ご注意ください。
1. 前提
1-1. 参照資料
以下のURLの資料を元にインストールします。
NVIDIAインストール
NVIDIA Driver Installation Quickstart Guide :: NVIDIA Tesla Documentation
https://docs.nvidia.com/datacenter/tesla/tesla-installation-notes/index.html
1-2. OS
以下のような環境にインストールします。
| 項目 | 詳細 |
|---|---|
| OS | Ubuntu 22.04.3 LTS |
| OSの状態 | インストール直後の状態 |
| CPU | AMD Ryzen 5 2400G with Radeon Vega Graphics |
| GPU | NVIDIA GeForce RTX 2070 |
2. 「1.1. Pre-Installation Requirements」を実行する
NVIDIA Driver Installation Quickstart Guideの1.1はインストール下準備です。
2.1. Verify You Have a CUDA-Capable GPU
CUDAが利用できるGPUであるかどうか確認します。
lspci | grep -i nvidia
こんな感じの結果が表示されます
01:00.0 VGA compatible controller: NVIDIA Corporation TU106 [GeForce RTX 2070] (rev a1)
01:00.1 Audio device: NVIDIA Corporation TU106 High Definition Audio Controller (rev a1)
01:00.2 USB controller: NVIDIA Corporation TU106 USB 3.1 Host Controller (rev a1)
01:00.3 Serial bus controller: NVIDIA Corporation TU106 USB Type-C UCSI Controller (rev a1)
2.2. Verify You Have a Supported Version of Linux
サポートされているLinuxディストリビューションかどうか確認します
uname -m && cat /etc/*release
こんな感じの結果が出力されます。
x86_64
DISTRIB_ID=Ubuntu
DISTRIB_RELEASE=22.04
DISTRIB_CODENAME=jammy
<<<後略>>>
2.3. Verify the System Has gcc Installed
gccがインストールされているかどうか確認します
gcc --version
こんな感じの結果が出力されます
gcc (Ubuntu 11.4.0-1ubuntu1~22.04) 11.4.0
Copyright (C) 2021 Free Software Foundation, Inc.
This is free software; see the source for copying conditions. There is NO
warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
2.4. Verify the System has the Correct Kernel Headers and Development Packages Installed
Kernel Headerパッケージが入っているかどうかという話ですが、パッケージでインストールする場合には自動的にインストールされるはずなので、無視して構いません。runfileでインストールする場合はインストールしておきましょう。
2.5. Install MLNX_OFED
GPUDirectStorage (GDS)を使わない場合は無視して構いません。
今回は使わないのでスキップします。
2.6. Choose an Installation Method
Linuxディストリビューションのパッケージでインストールすることをお勧めしますよ、と書いているだけなのでスキップします。
2.7. Download the NVIDIA CUDA Toolkit
NVIDIA CUDA Toolkitをダウンロードします。
ダウンロードするURLは以下です。
今回は、OS-->「Linux」、Arch-->「x86_64」、Dist-->「Ubuntu」、Version-->「22.04」、Installer Type-->「deb(network)」を選択しました。
出力されたインストラクションは以下の通りです。
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/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
上記を実行します。
cuda-driverのみインストールしたい場合は、
sudo apt-get -y install cuda-drivers
としてください。
エラーがなくインストール終了したらOKです。
3. インストールの確認
3-1. CUDAインストール確認
リブートする前に何がインストールされたかcuda関連パッケージを確認します
以下のコマンドでインストールされたcuda関連パッケージを確認できます。
apt list --installed cuda\*
必要なCUDA DriverやToolkitがインストールされていること確認しましょう
3-2. NVIDIAドライバーインストール確認
以下のコマンドでインストールされたNVIDIAドライバーを確認できます。
sudo apt list --installed nvidia\*
4. 再起動
リブートします
sudo reboot
5. 再起動後の確認
5-1. NVIDIAドライバーの確認
ディストリビューションのデフォルトnouveauドライバーが無効になっているか確認します。
lsmod | grep nouveau
何も検索結果が出力されなければOKです。
あちこちで書かれているNVIDIAドライバーインストール記事では、nouveauドライバーを無効にすると書かれていますが、本記事の方法では無効にする必要はありません。自動的に無効になります。
NVIDIAドライバーに切り替わっているか確認します。
lsmod | grep nvidia
以下のように出力されたら切り替わっています
nvidia_uvm 1761280 0
nvidia_drm 90112 3
nvidia_modeset 1314816 6 nvidia_drm
nvidia 56619008 210 nvidia_uvm,nvidia_modeset
drm_kms_helper 249856 1 nvidia_drm
drm 696320 7 drm_kms_helper,nvidia,nvidia_drm
i2c_nvidia_gpu 16384 0
i2c_ccgx_ucsi 16384 1 i2c_nvidia_gpu
video 73728 2 asus_wmi,nvidia_modeset
5-2. NVIDIA-SMIの実行
nvidia-smi
こんな感じの結果が出力されます
6. まとめ
NVIDIAドライバーインストールは様々な記事がインターネットには溢れていて、情報が錯綜していますが、恐らく上記の手法が最速最短の方法だと筆者は思っております。もっと簡単な方法がありましたら、コメントにお寄せいただくと幸いです。
7. 謝辞
最も簡単にNVIDIAドライバーをインストールする方法をお教えいただいたNVIDIA佐々木様に感謝いたします。