はじめに
この記事ではUbuntuでGPUを動かすためにCuPyをインストールするまでを備忘録として記録します。
ゴールは以下のとおりです。
-
Ubuntu 20.04にNVIDIA driverをインストール
-
CUDA Toolkit 11.1をインストール
-
CuPy 11.1をインストール
実行環境
-
OS: Ubuntu 20.04
-
GPU: RTX 2060 SUPER
-
CPU: Ryzen 3600
私がやりたいこととしては、poetryを使った仮想環境を作り、その中に諸々をインストールしますが、ここでは割愛します。
インストール方法
1. Ubuntu 20.04にNVIDIA driverをインストール
以下のコマンドで、推奨ドライバーを見つける。
ubuntu-drivers devices
ソフトウェアとアップデートの追加ドライバーから推奨されたドライバーをインストールできます。
もしくは、以下のコマンドでNVIDIAのPPAリポジトリを追加してドライバーをインストールします。
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
sudo apt install nvidia-driver-xxx
（コメント: もしかしたらこの辺が逆かもしれない。）
ドライバーがインストールできたら以下のコマンドで確認できます。
nvidia-smi
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 460.27.04 Driver Version: 460.27.04 CUDA Version: 11.2 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 GeForce RTX 206... On | 00000000:06:00.0 On | N/A |
| 29% 31C P3 39W / 175W | 1114MiB / 7979MiB | 19% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
| 0 N/A N/A 1104 G /usr/lib/xorg/Xorg 101MiB |
| 0 N/A N/A 3110 G /usr/lib/xorg/Xorg 586MiB |
| 0 N/A N/A 3271 G /usr/bin/gnome-shell 134MiB |
| 0 N/A N/A 4748 G ...AAAAAAAA== --shared-files 212MiB |
+-----------------------------------------------------------------------------+
（コメント: できない場合は再起動する。）
2. CUDA Toolkit 11.1をインストール
CUDA ToolkitとCuPyはバージョンを合わせる必要がある（はず）ので、自分に必要なバージョンのCUDA Toolkitをインストールします。
今回は現時点（2021/1/8）で最新であるCuPy 11.1を入れたいので、それに合わせてCUDA Toolkitも11.1をインストールします（この時点で最新なバージョンは11.2）。
（コメント: 先程のコマンドの表示で右上にCUDA Version: 11.2と表示していますが、これは入れられるバージョンだそうです）
過去のバージョンはCUDA Toolkit Archiveにあります。
CUDA Toolkit 11.1.1 -> Linux -> x86_64 -> Ubuntu -> 20.04 -> runfile (local)
の順で進みます。
下に示されるコマンドに従い、インストールします。
wget https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run
sudo sh cuda_11.1.1_455.32.00_linux.run
ここで、Continue -> accept と進みます。
CUDA Installerでは、Driverのチェックを外します（エンターキーで外れます）。外したあとにInstallへ進みます。
（コメント: Driverをインストールしてしまうとエラーを起こします。）
3. CuPy 11.1をインストール
公式サイトを参考に自身の仮想環境なりで以下のコマンドでインストールします。
pip install cupy-cuda111
おわりに
いろいろ試行錯誤した結果、この方法が一番ラクだった気がします。