検証環境
| OS | GPU |
|---|---|
| Ubuntu20.04(x86_64 | Quadro M4000 |
Linuxなら大体同じ方法で大丈夫と思います
ステップ
- ドライバのインストール
- cuDNNのインストール
- Cuda Toolkitのインストール
GPUの確認
GPUの確認
lspci | grep NVIDIA
# 1:00.0 VGA compatible controller: NVIDIA Corporation TU106 [GeForce RTX 2060 SUPER] (rev a1)
CPUアーキテクチャの確認
CPUアーキテクチャの確認
uname -p
OSディストリビューションの確認
OSディストリビューションの確認
lsb_release -d
ドライバインストール
aptから取得できますが、バージョンが複数あります。
バージョンが複数ある
apt-cache search nvidia | grep nvidia-driver | grep -v server
まず対応するバージョンをここで確認
インストールするマシンのOSやGPUの種類をを選んで[Search]を押してください
ここでダウンロードできる.runファイルを実行してもいいらしいですが自分はaptでやりました
今回はバージョン470なので
バージョン470インストール
apt-get install nvidia-driver-470
aptでドライバが見つからない場合
ここからダウンロードした.runファイルを実行してインストールできる。
novelu-kernelのエラーが出る場合は指示に従ってinitramfsを更新後再起動して再度実行する
initramfsを更新
sudo update-initramfs -u
reboot
CUDA Toolkitのインストール
ここの指示にしたがってレポジトリ追加してapt-getしてください
CUDA Toolkitのインストール
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-4-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
cuDNNのインストール
runtime libraryを選びました
.debか.tarで選べるみたいですが.tarの方にしました
cuDNNのインストール
tar xvf cudnn-11.4-linux-x64-v8.2.4.15.tgz
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
あとはパスを通してください
パスを通す
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH}
インストール後にGUIで起動するようになってしまう問題
CUI環境でNVIDIAドライバをインストールすると、自動的にXがインストールされGUIで起動するようになってしまいます。
自分はUbuntu Serverを使っており常時稼働させており、デフォルトで20分経過後にサスペンドされてしまい大変不便でした。
以下CUIブートへの切り替え手順です。
CUIブートへの切り替え
systemctl get-default
grapycal.target #GUIで起動する状態
#CUIで起動するように変更
systemctl set-default multi-user