nVIDIAドライバのインストール

ppaを追加してドライバをインストールします。

$ sudo add-apt-repository ppa:graphics-drivers
$ sudo apt-get update

本来はCUIでインストールを進めるらしいです。

$ sudo service stop lightdm

apt installでインストールです。

$ sudo apt install nvidia-driver-455 nvidia-settings

cudaインストール

たぶん、こんな感じ・・・

$ 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.1.0/local_installers/cuda-repo-ubuntu2004-11-1-local_11.1.0-455.23.05-1_amd64.deb
$ sudo dpkg -i cuda-repo-ubuntu2004-11-1-local_11.1.0-455.23.05-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-ubuntu2004-11-1-local/7fa2af80.pub
$ sudo apt update
$ sudo apt -y install cuda

成功してるとnvidia-smiで次のように表示できます。

$ nvidia-smi 
Tue Oct 27 21:38:32 2020       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 455.23.05    Driver Version: 455.23.05    CUDA Version: 11.1     |
|-------------------------------+----------------------+----------------------+
| 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 3080    On   | 00000000:01:00.0 Off |                  N/A |
| 53%   31C    P3    65W / 370W |    654MiB / 10018MiB |      5%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      1435      G   /usr/lib/xorg/Xorg                515MiB |
|    0   N/A  N/A      2518      G   ...AAAAAAAAA= --shared-files      137MiB |
+-----------------------------------------------------------------------------+

追加

$ sudo ln -s /usr/lib/x86_64-linux-gnu/libGLU.so.1 /usr/lib/libGLU.so
$ sudo apt install freeglut3 freeglut3-dev

デモをビルドする。

$ /usr/local/cuda-11.1/bin/cuda-install-samples-11.1.sh ./
$ cd NVIDIA_CUDA-11.1_Samples/
$ make