CUDA Toolkit 및 Cudnn 설치

ubuntu 16.04 

 

CUDA Toolkit 설치 

 

1. Nvidia 드라이버는 미리 설치해야함

 

2. 아래 링크를 통해 본인 환경을 선택 (본인은 Installer Type : deb(local)으로 선택함)

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

3. Base Installer에 나와 있는 Installation Instructions을 따른다. 

 

4. 모든 코드를 설치하고 환경 변수를 설정하여 준다.

 

$nano ~/.bashrc 

export PATH=/usr/local/cuda/bin:/$PATH <- 해당 내용 추가

5. sudo reboot를 통한 서버 재부팅을 진행한 후 nvcc -V를 통해 설치한 CUDA 버전 확인

 

Cudnn 설치 

$ tar –xzvf cudnn-(본인 Cudnn 버전에 맞는 내용 기입 *일반적으로 tap키를 누르면 해결) 
$ 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*

끝