OS & Tools/Linux

[ubuntu 16.04] CUDA & CuDNN 설치

전두선 2020. 2. 4. 18:30

저는 이 포스터를 작성하면서 CUDA 10.1 & CuDNN 7.6.4 환경을 구성하였습니다. 다른 버전을 설치하는 분들 께서는 이를 참고해가지고 설치를 진행해주세요. (중간중간 경로명 ex> cuda-10-2 -> cuda-9.0 그리고 설치파일명 부분만 신경써서 바꿔주면 다를게 없이 쉽게 설치가능)

 

> CUDA 

1. CUDA Toolkit 설치파일 다운로드

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

 

2. CUDA Toolkit 10.1 ver

 

저는 CUDA 10.1 버전을 설치하고자 합니다. 해당 다운로드 사이트로 접속하면 아래와 같은 인터페이스가 있는데, 자신의 컴퓨터 사양에 맞게 옵션을 선택해주면 됩니다.

installer Type은 deb(local)을 눌러줍니다.

 

CUDA Toolkit 10.2

그렇다면 하단에 설치 자료를 제공해주는데, 따라서 진행하면 설치 끝.

 

 

> 환경 변수 설정

$ echo -e "\n## CUDA and cuDNN paths" >> ~/.bashrc
$ echo 'export PATH=/usr/local/cuda-10.1/bin:${PATH}' >> ~/.bashrc
$ echo 'export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:${LD_LIBRARY_PATH}' >> ~/.bashrc
$ source ~/.bashrc

 

> 설치 완료 및 확인

$ nvcc --version
$ which nvcc # /usr/local/cuda-10.1/bin/nvcc
/usr/local/cuda-10.1/bin/nvcc
여기서 'which nvcc' 명령을 입력했을 때, 결과가 나오지 않는다면 이전 과정의 환경변수 설정 혹은 CUDA 설치 과정에서 문제가 발생한 것이므로 다시 돌아가서 점검해야 합니다.

 

> CuDNN 

https://developer.nvidia.com/rdp/cudnn-archive

저같은 경우는 위에서 CUDA 10.1을 설치했으므로 그에 맞는 7.6.4 버전의 CuDNN을 설치합니다. CUDA 버전에 따라, 우분투 버전에 따라 설치 파일이 다르므로 잘 체크하여 다운 받습니다. 

 

$ cd Downloads
$ tar xzvf cudnn-10.1-linux-x64-v7.6.4.38.tgz
$ which nvcc
/usr/local/cuda-10.1/bin/nvcc
$ sudo cp cuda/lib64/* /usr/local/cuda-10.1/lib64/
$ sudo cp cuda/include/* /usr/local/cuda-10.1/include/
$ sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*
$ sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h

> 설치 완료 및 확인

아래 명령어를 입력하여 에러가 아닌 버전이 표시되면 설치가 이상없이 된 것입니다.

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

#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 4
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"

 

> 기타 패키지 설치

$ sudo apt-get install libcupti-dev

> 기존에 설치된 CUDA, CuDNN만 제거

$ sudo apt-get --purge remove 'cuda*'
$ sudo apt-get autoremove --purge 'cuda*'

# /usr/local/에 cuda 또는 cuda-xx.x 폴더가 있다면 지워준다.

 

 

> 기존에 설치된 CUDA, CuDNN, 그래픽 드라이버까지 전부 다 제거

$ sudo apt-get purge nvidia* && sudo apt-get autoremove && sudo apt-get autoclean && sudo rm -rf /usr/local/cuda*
$ sudo reboot now

 

1 2 3 4 5 6