Ubuntu (우분투) CUDA, cuDNN 설치

아래 링크에서 CUDA 11.8 및 cuDNN 8.9.5를 다운받을 수 있다.

https://drive.google.com/drive/folders/1pY23ehSFqRqn-abC_mwkdSy4QAygKjik?usp=sharing

CUDA 설치

아래 링크에서 CUDA 11.8 버전을 다운로드 받을 수 있다. 

https://developer.nvidia.com/cuda-11-8-0-download-archive

 

Ubuntu CUDA 설치파일을 받기 위해서 아래 사진과 같이 선택한다. 

아래와 같은 명령어로 CUDA를 우분투에 설치할 수 있다.

$ wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
$ sudo bash cuda_11.8.0_520.61.05_linux.run

이미 Nvidia driver를 설치하면 아래와 같은 화면이 나온다. Continue로 이동해서 무시하면 된다.

이미 Driver를 설치했으면 Driver에 ENTER 눌러서 X표시를 제거하고 Install 하도록 하자.

설치가 완료 되면 /usr/local/ 경로에 cuda와 cuda-11.8 폴더 생긴 걸 확인할 수 있다.

---

cuDNN 설치

아래 링크에서 cuDNN을 다운받을 수 있다.

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

CUDA 11.x버전이므로 아래와 같이 선택해서 cuDNN을 다운받아야 한다. (여기서 Tar 포맷의 설치파일을 받도록 하자.)

cuDNN을 다운 받으려면 NVIDIA에 로그인해야 한다.

아이디가 없는 경우, 아래 링크로 CUDA 11.x 전용 cuDNN 설치파일을 받을 수 있다. 

https://developer.nvidia.com/downloads/compute/cudnn/secure/8.9.5/local_installers/11.x/cudnn-linux-x86_64-8.9.5.29_cuda11-archive.tar.xz/

아래 명령어로 설치파일 압축을 해제하고 CUDA폴더로 복사하면 된다.

$ tar -xvf cudnn-linux-x86_64-8.9.5.29_cuda11-archive.tar.xz
$ cd cudnn-linux-x86_64-8.9.5.29_cuda11-archive
$ sudo cp include/cudnn* /usr/local/cuda-11.8/include
$ sudo cp lib/libcudnn* /usr/local/cuda-11.8/lib64/
$ sudo chmod a+r /usr/local/cuda-11.8/lib64/libcudnn*

다음과 같이 cuDNN이 정상적으로 설치된 것을 확인할 수 있다.

$ cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR 8
#define CUDNN_MINOR 9
#define CUDNN_PATCHLEVEL 5

 

참고

 

1. https://dohyeon.tistory.com/14