Ubuntu 16.04에 CUDA Toolkit과 CuDNN 설치하기

연구실 1080TI 2대가 있는 서버에 CUDA 세팅을 하게되었다.

위의 과정은 크게 3가지로 구분된다.

 

- Nvidia Driver 설치

- CUDA Toolkit 설치

- CuDNN 설치

 

 

Nvidia Driver 설치

sudo apt-get update
ubuntu-drivers devices

위의 명령어를 사용해 nvidia driver 몇 버전을 설치해야 하는지 확인한다.

nvidia-430을 추천하라기에 설치했다.

sudo apt-get install nvidia-430

 

nvidia driver 설치할때 다양한 이슈가 있었다. 

가장 큰 이슈는 기존에 ubuntu에서 사용하는 noveau라는 driver를 중단시키고 설치해야 한다는 것이다.

nvidia driver를 설치한 이후에는 noveau를 다시 실행시켜야 ubuntu UI를 활용할 수 있다.

 

설치 중 기존 드라이버와 충돌시에는 아래 명령어를 사용해 nvidia 드라이버를 삭제한다.

 

apt --purge autoremove nvidia*

 

설치 후에는 nvidia-smi를 실행시켜 driver를 사용할 수 있는지 확인한다.

 

 

CUDA Toolkit 설치

Toolkit 설치 전 Dependency를 설치한다.

apt-get update
apt-get install build-essential dkms
apt-get install freeglut3 freeglut3-dev libxi-dev libxmu-dev

 

Nvidia site에 가서 버전에 호환되는 toolkit을 설치한다.

버전은 nvidia-smi를 사용해 CUDA 버전을 확인할 수 있다.

https://developer.nvidia.com/cuda-toolkit-archive

CUDA Toolkit Archive

CUDA Toolkit을 설치한 후에 환경변수를 잡아준다.

export PATH=$PATH:/usr/local/cuda-11.0/bin:$PATH
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH

nvcc --version 코드를 실행시키면 CUDA Toolkit이 제대로 설치되었는지 확인할 수 있다.

 

 

CuDNN 설치

CuDNN 설치가 제일 애매했다.

위의 그림의 노란색 부분 세개를 반드시 모두 설치해야 CuDNN이 설치가 된다.

설치후 실행 명령어는 다음과 같다.

$ tar -xzvf cudnn-10.1-linux-x64-v7.6.5.32.tgz
$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

$ sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.1_amd64.deb
$ sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.1_amd64.deb