Ubuntu18.04にNVIDIAドライバをインストールする方法 (備忘録)

NVIDIAからダウンロードしてきたGeForceドライバ(Ubuntu上でのNVIDIA Driver)、もしくはCUDA Toolkit(NVIDIAドライバ含む)をrunファイルからインストールする場合、まずカーネルモジュール[nvidia-drm]をアンロードしてXserverを止めろと言ってきてエラーになり必要なNVIDIA Driverがインストールできない。例としてNVIDIA CUDA(デフォルト設定ではNVIDIA Driverが自動でインストールされるようになっている)のインストール手順を備忘録として書き残します。

CUDA Toolkitからでなく、GeForceドライバ(Ubuntu上でのNVIDIA Driver)単体でインストールする場合は次のページがとても参考になります。

[ UEFI のセキュアブート機にNVIDIAのドライバを入れる話 ]

OS : Ubuntu 18.04
CUDAバージョン : 11.2
CUDAファイル名(※) : cuda_11.2.0_460.27.04_linux.run

(※)ファイル命名規則 : cuda _ [CUDAバージョン] _ [NVIDIAドライババージョン] _ [OS種別].run

手順１ : NVIDIAのサイトからCUDA Toolkitをダウンロードしておく。

ここでは~/Downloads/ディレクトリにダウンロードしたものとして手順を書きます。

手順２ : CUI(マルチユーザー)モードで起動するための前準備１

設定を失敗するとログイン画面が表示されなくなる可能性があるため、注意しながら設定してください。

$ cd /lib/systemd/system
$ sudo unlink default.target
$ sudo ln -s multi-user.target default.target

手順３ : CUI(マルチユーザー)モードで起動するための前準備２

$ ls -l
(↓のようになっていればOK)
lrwxrwxrwx 1 root root 16  1月  3 01:46 default.target -> multi-user.target

手順４ : 再起動

$ reboot

うまくいっていればCUIモードで起動するので、ユーザー名とパスワードを入力しログインする。

手順５ : CUDA Toolkitのインストール

$ cd ~/Downloads
$ sudo sh cuda_11.2.0_460.27.04_linux.run

インストール画面についてはメニュー形式で表示されるため省略します。
(オプションもあるので、必要に応じて選択してください)

ちなみにインストール中、画面が固まったように見えることがありますが、Toolkitのファイルサイズが大きいため、処理に時間がかかっているだけです。放っておけば通常は処理が完了します。

手順６ : GUIモードで起動するための前準備１

設定を失敗するとログイン画面が表示されなくなる可能性があるため、注意しながら設定してください。

$ cd /lib/systemd/system
$ sudo unlink default.target
$ sudo ln -s graphical.target default.target

手順７ : GUIモードで起動するための前準備２

$ ls -l
(↓のようになっていればOK)
lrwxrwxrwx 1 root root 16  1月  3 01:46 default.target -> graphical.target

手順８ : 再起動

$ reboot

手順９ : CUDA Toolkitのパスを通す

インストール完了画面に次の２つのディレクトリにパスを通すよう書かれています。
/usr/local/cuda-11.2/lib64
/usr/local/cuda-11.2/bin

ディレクトリにcuda-11.2と書くように指定されていますが、cuda-11.2ディレクトリにcudaというシンボリックリンクが作られているので、以下のexport文のように~/.bashrcファイルに記述すればOKです。

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

さいごに、sourceコマンドで.bashrcを読み込み直せば作業は完了です。

$ source ~/.bashrc

自分の備忘録なのでざっくりしか書いていませんが、この手順がわからずにNVIDIA DriverやCUDA Toolkitのインストールを諦めている人もいるかもしれないので、少しでも参考になれば幸いです。

参考にしたサイト
CentOS7、Ubuntu 18.04 でGUIログインを無効にするには
UEFI のセキュアブート機にNVIDIAのドライバを入れる話