はじめに
2018年11月現在、PyTorchが正式にCUDA 10.0に対応していないため、CUDA 9.2をUbuntu 16.04にインストールする。このときに、NVIDIA GeForce RTX 20シリーズを載せていると問題が起きたため、解決方法を記載。
症状
NVIDIA GeForce RTX 20シリーズではDriver 410以降が必要。しかし、Driver 410をインストールしたあとに、CUDA 9.2をインストールすると強制的にDriver 396に下げられる。Driver 396ではGeForce RTX 20シリーズに対応していないため、ログインループに陥ってしまう。
環境
- Ubuntu 16.04.5 LTS (Xenial Xerus)
- NVIDIA GeForce RTX 2070
- Bash 4.3.48
解決手順
大まかな流れ
-
Driverをインストールせずに、CUDA 9.2をインストール
-
CUDAと同時に自動的にインストールされたDriverを削除
-
PPAからDriver 410をインストール
詳細な手順
1. はじめはNVIDIA Driverはインストールしない。
もし、インストールしている場合は以下のようにして、削除する。
terminal
$ sudo apt --purge remove nvidia-* # Driverの削除
$ sudo apt --purge remove cuda-* # CUDAの削除
$ sudo apt autoremove # 不要になったファイルの削除
また、ログインループに陥っているときはctrl + alt + F1でCUIでログインしてから、上記のコマンドを実行する。
ctrl + alt + F7でGUIに戻れる。
2. CUDA 9.2をダウンロード
CUDA 9.2を下記サイトからダウンロード
https://developer.nvidia.com/cuda-92-download-archive
このとき、Installer Typeは「deb(local)」を選択。
3. CUDA 9.2をインストール
(例)
terminal
$ cd ~/Downloads
$ sudo dpkg -i cuda-repo-ubuntu1604-9-2-local_9.2.148-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-9-2-local/7fa2af80.pub
$ sudo apt update
$ sudo apt install cuda
4. CUDAインストールと同時に自動的にインストールされたDriverを削除
terminal
$ dpkg -l | grep nvidia-* # これでインストール済みのパッケージが表示される。
$ sudo apt --purge remove nvidia-* # Driverの削除
$ dpkg -l | grep nvidia-* # これで何も表示されなければ削除成功
5. CUDAのパスを設定
.bashrcなどに以下を追記
.bashrcなどに追記
# CUDA
export PATH=/usr/local/cuda-9.2/bin:${PATH}
export LD_LIBRARY_PATH=/usr/local/cuda-9.2/lib64:${LD_LIBRARY_PATH}
6. PPAの追加
terminal
$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt update
7. Driver 410のインストール
terminal
$ sudo apt install nvidia-410
8. 再起動
terminal
$ reboot