CUDA 설치

$ cd ~/Downloads
$ wget <https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.2.89-1_amd64.deb>
$ sudo dpkg -i cuda-repo-ubuntu1804_10.2.89-1_amd64.deb
$ sudo apt-key adv --fetch-keys <https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub>
$ sudo apt-get update
$ sudo apt install cuda
$ sudo apt install cuda-10-1

cuDNN 설치

$ cd ~/Downloads
$ wget <https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb>
$ sudo dpkg -i nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
$ sudo apt update
$ sudo apt install libcudnn7-dev
$ sudo apt install libnvinfer-dev=6.0.1-1+cuda10.1 libnvinfer6=6.0.1-1+cuda10.1 libnvinfer-plugin6=6.0.1-1+cuda10.1

설치 확인 (재부팅후)

$ nvidia-smi 
Sun Jun  7 11:57:04 2020       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 450.36.06    Driver Version: 450.36.06    CUDA Version: 11.0     |
|-------------------------------+----------------------+----------------------+
| 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 GTX 105...  On   | 00000000:01:00.0 Off |                  N/A |
| N/A   64C    P0    N/A /  N/A |   1156MiB /  4042MiB |      3%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      1166      G   /usr/lib/xorg/Xorg                 72MiB |
|    0   N/A  N/A      1397      G   /usr/bin/gnome-shell               50MiB |
|    0   N/A  N/A      1983      G   /usr/lib/xorg/Xorg                557MiB |
|    0   N/A  N/A      2176      G   /usr/bin/gnome-shell              255MiB |
|    0   N/A  N/A      9437      G   ...AAAAAAAAA= --shared-files      218MiB |
+-----------------------------------------------------------------------------+