[Ubuntu] Install CUDA

0. Overview

 

Nvidia의 인공지능 병렬처리를 위해서

다음이 필요하다.

 

1. Nvidia Driver
2. CUDA
3. cuDNN

또한, 이들 간에 버전을 서로 맞춰주어야한다.

 

위 사진은

cuDNN 8.2.2와 호환되는 CUDA와 Driver의 버전이다.

 

아래의 링크에서 다른 cuDNN버전에 대해서도 확인가능하다.

https://docs.nvidia.com/deeplearning/cudnn/support-matrix/index.html#fntarg_1

Support Matrix :: NVIDIA Deep Learning cuDNN Documentation

 

 

 

 

1. Install CUDA

 

CUDA는 아래의 링크에서 설치가능하다.

https://developer.nvidia.com/cuda-downloads

CUDA Toolkit 11.4 Update 1 Downloads

Select env.

 

Example : CUDA 11.4.1

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
$ sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
$ wget https://developer.download.nvidia.com/compute/cuda/11.4.1/local_installers/cuda-repo-ubuntu1804-11-4-local_11.4.1-470.57.02-1_amd64.deb
$ sudo dpkg -i cuda-repo-ubuntu1804-11-4-local_11.4.1-470.57.02-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-ubuntu1804-11-4-local/7fa2af80.pub
$ sudo apt-get update
$ sudo apt-get -y install cuda