cuda & cudnn 설치

gpu 세팅을 위한 cuda와 cudnn 설치 방법에 대하여 설명하고자 한다.

 

cuda 사이트에서 적합한 파일을 선택하여 설치하면 된다.

 

Linux > x86_64 > Ubuntu > 16.04 > deb(local)을 설치했으며,

 

선택 후 나타나는 코드를 실행하면 기본적으로 cuda는 잘 설치되었다고 볼 수 있다.

 

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

CUDA Toolkit 11.1 Downloads

하지만 여기에서 설치 경로를 지정해주어야 한다.

$ export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}

$ export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

$ source ~/.bashrc

$ nvcc --version

위와 같이 경로 지정까지 마치면 nvcc --version에서 cuda버전이 나타나는 것을 볼 수 있다.

 

cuda를 설치한 이후, cudnn을 설치해주어야 한다.

 

cudnn의 드라이브는 nvidia사이트에서 받을 수 있다.

 

developer.nvidia.com/rdp/cudnn-archive

cuDNN Archive

 

$ cd 설치경로

$ sudo tar -xzvf cudnn-10.1-linux-x64-v7.6.4.38.tgz 

$ cd cuda

$ sudo cp include/cudnn.h /usr/local/cuda/include

$ sudo cp lib64/libcudnn* /usr/local/cuda/lib64

$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

 

위의 nvidia사이트에서 cudnn에 대한 파일을 받고 코드를 입력하면 마지막에  CUDNN_MAJOR가 뜨는 것을 확인할 수 있다.

 

 

출처:

https://hwiyong.tistory.com/86

[Ubuntu 16.04] CUDA + Cudnn 설치하기