[ubuntu] 18.04 CUDA, cuDNN 설치

이 글은 2021년 8월 9일 기준 작성되었습니다.

시간이 지나면 버전이 달라지면서 에러가 발생할 수 있습니다. 

 

1. CUDA 설치 

---

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

CUDA Toolkit Archive

 

해당 링크에서 본인에게 맞는 CUDA version을 선택해서 설치

 

1.1 아래 명령어를 순서대로 실행

wget https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run

sudo sh cuda_11.0.3_450.51.06_linux.run

 

• 드라이버가 이전에 설치되어 있다고 뭐라뭐라 하는데 무시하고 continue 선택하고 enter 
• "Do you accept the above EULA?" 라는 명령어가 나오면 accept 입력하고 enter

 

 

• 해당 창에서 gpu driver는 앞서 이미 설치했기 때문에 Driver 선택하고 enter 누르면 x 체크박스가 해제됩니다. 
• 해제후 Install 
• 다음과 같은 메세지가 뜨면 성공

driver 는 선택되지 않았고, Toolkit은 /usr/local/cuda-11.0 에 설치되었고, CUDA samples 파일은 /root에 설치되었습니다. 

 

1.2 CUDA PATH 설정 

gedit ~/.bashrc

bashrc 파일에 들어가서 아래 3줄을 cuda version 본인의 것에 맞게 고치고 bashrc 파일 맨 밑줄에 넣어준다

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

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

export LC_LIBRARY_PATH=/usr/local/cuda-11.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH

save 누르고 파일 종료 후 아래 명령어 입력

source ~/.bashrc

위를 통해 설정한 PATH를 현재 접속해 있는 터미널에 적용 

 

1.3 설치 확인

nvcc -V

다음과 같이 11.0 version이 성공적으로 설치되었습니다.

 

2. cuDNN 설치

---

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

해당 링크에서 CUDA 버전에 맞는 cuDNN을 설치 (회원가입 후 로그인 해야만 다운로드 가능) 

'cuDNN Library for Linux[x86_64]' 설치

다운로드 받은 파일을 바탕화면에 옮기고 터미널 실행 

실행 후 

#바탕화면으로 이동
cd /home/csdl/Desktop

#압축해제 
tar -xzvf cudnn-11.0-linux-x64-v8.0.5.39.tgz

결과적으로 바탕화면에  cuda 폴더 생성됨

 

폴더 안으로 들어가보면 include 폴더, lib64 폴더, 어떤 txt 파일이 하나 생성되어있음

include 폴더안의 모든 파일은 cuda의 include폴더에 복사해주고, 
lib64 폴더안의 모든 파일은 cuda의 lib64로 복사해주면 됨 

 

cd cuda 
sudo cp include/cudnn* /usr/local/cuda-11.0/include
sudo cp lib64/libcudnn* /usr/local/cuda-11.0/lib64/
sudo chmod a+r /usr/local/cuda-11.0/lib64/libcudnn*

 

2.1 cuDNN 설치 확인

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

다음과 같이 코드가 뜨면 성공~~!