はじめに

UbuntuでNDIVIAのGPUマシーンを構築する手順を解説します。 GPUの装着やOSのセットアップは完了していることを前提としています。
自前でGPUマシーンを構築する場合、NDIVIAのドライバなどをインストールする必要があります。

環境調査

今回のPC環境は以下の通り。

OSのバージョンは以下のコマンドでも確かめられる。

cat /etc/os-release

PCIデバイスの確認はこちら

lspci | grep -i nvidia

デフォルトでは'nouveau'というドライバーが動作している。以下のコマンドで確かめられる。

lsmod | grep nouveau

以下のコマンドでも確認可能。Linux カーネルが起動したときに出力したメッセージ。

dmesg | grep -i nouveau

NVIDIA GPUのモデルと推奨ドライバーを検出する。今回はrecommendedとある、'nvidia-driver-515'にする。

ubuntu-drivers devices

今回は、Ubuntuの公式リポジトリからNVIDIAドライバをインストールする。NVIDIAのサイトからインストールすることも可能。

NVIDIAドライバを検索する。

apt-cache search nvidia-driver-515

NVIDIAドライバをインストール。

sudo apt-get install nvidia-driver-515

OSを再起動。

systemctl reboot

OS再起動後、再度dmesgでnouveauドライバを検索し、何も出力されなければOK。

lsmod | grep nouveau

NVIDIAドライバがロードされていることを確認する。

lsmod | grep nvidia

NVIDIAの管理インターフェースで、GPUの状態を確認。

nvidia-smi

カーネル起動時のメッセージにはエラーが表示されるが、調べても有効な解決策は無さそう。

dmesg | grep -i nvidia