Ubuntu에 CUDA, CuDNN 설치하기

* 본 포스팅은 RTX 3070, Ubuntu 18.04를 기준으로 작성합니다.

1. CUDA

CUDA ("Compute Unified Device Architecture", 쿠다)는 그래픽 처리 장치(GPU)에서 수행하는 (병렬 처리) 알고리즘을 C 프로그래밍 언어를 비롯한 산업 표준 언어를 사용하여 작성할 수 있도록 하는 GPGPU 기술이다.

 

사용하고 있는 GPU에 맞는 CUDA 버전을 확인해줍니다.

en.wikipedia.org/wiki/CUDA#GPUs_supported

CUDA - Wikipedia

 

제가 사용하는 GPU는 RTX 3070이므로 Compute capabality는 8.6입니다.

Compute capabality 7.5는 CUDA11버전을 지원합니다. 본 글에서는 11.1 버전으로 진행합니다.

developer.nvidia.com/cuda-toolkit-archive

CUDA Toolkit Archive

본인의 운영체제에 맞는 옵션을 선택해줍니다.

Base Installer에 커맨드가 나왔으면 창을 놔두고 terminal 창을 띄워줍니다.

2. CUDA 설치

우선 터미널에서 다음 명령어를 실행하여 라이브러리의 업데이트를 실행해줍니다

sudo apt-get update

Base Installer의 첫번째줄을 복사해 커맨드에 입력해줍니다. 설치 파일을 다운로드 하는 과정입니다.

wget https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run

-위 스크린샷과 별개로 11.1.1 버전으로 포스팅을 진행합니다!

아래 커맨드로 설치파일을 실행시켜줍니다.

sudo sh cuda_11.1.1_455.32.00_linux.run

accept를 입력 후 Enter를 눌러줍니다.

드라이버까지 일괄로 설치할 경우는 모든 칸에 X표시가 체크되어 있는 상태로 Install 해주시면 됩니다. 저는 드라이버는 따로 설치하기 위해 Driver칸 체크를 해제 후 진행했습니다. 체크를 해제하지 않은 경우는 다음 드라이버 설치 과정을 생략하셔도 됩니다. 

오류없이 마무리가 되었으면 설치가 완료된것입니다.

3. Nvidia-Driver 설치

- CUDA 설치과정에서 드라이버를 설치하신 경우 본 단락은 생략하셔도 됩니다.

- 본 포스팅에서는 460버전을 설치합니다.

 

기존에 설치된 드라이버를 삭제해줍니다.

sudo apt --purge autoremove nvidia*

driver를 설치할 repository를 추가합니다.

sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt update

설치할 버전을 모르는 경우 아래 커맨드로 설치가능한 드라이버 목록을 출력합니다.

apt-cache search nvidia

apt로 드라이버를 설치해줍니다.

sudo apt-get install nvidia-driver-460

설치가 완료되면 재부팅해줍니다.

sudo reboot

4. 환경변수 등록

환경변수 등록을 위해 .bashrc 파일을 열어줍니다.

sudo vi .bashrc

.bashrc 파일에 다음과 같이 입력주 저장해줍니다.

export CUDA_HOME=/usr/local/cuda export LD_LIBRARY_PATH=${CUDA_HOME}/lib64 PATH=${CUDA_HOME}/bin:${PATH} export PATH

재부팅 후 아래 커맨드로 설치가 잘 되었는지 확인해줍니다.

nvidia-smi

5. cuDNN 설치

cuDNN은 엔비디아 CUDA 딥 뉴럴 네트워크 라이브러리, 즉 딥 뉴럴 네트워크를 위한 GPU 가속화 라이브러리의 기초 요소로 컨볼루션(Convolution), 풀링(Pooling), 표준화(Nomarlization), 활성화(Activation)와 같은 일반적인 루틴을 빠르게 이행할 수 있도록 하는 라이브러리입니다.

developer.nvidia.com/CUDNN

NVIDIA cuDNN

위 사이트에서 Download cuDNN -> Login -> License agree 후 CUDA 버전에 맞는 cuDNN을 받으시면 됩니다.

저는 8.0.5 for CUDA 11.1 로 진행합니다. cuDNN Library for Linux (x86_64)을 다운받아줍니다.

cuDNN을 다운로드 받은 폴더로 이동해준 후 압축파일을 풀어줍니다.(보통 Downloads 폴더)

cd Downloads tar -xvf cudnn-11.1-linux-x64-v8.0.5.39.tgz

압축을 풀면 cuda라는 폴더가 생기고, 이 안의 파일들을 cuda 설치폴더로 복사해주면 됩니다.

sudo cp ./cuda/include/* /usr/local/cuda-11.1/include sudo cp -P ./cuda/lib64/* /usr/local/cuda-11.1/lib64 sudo chmod a+r /usr/local/cuda-11.1/lib64/libcudnn*

아래 커맨드로 cuDNN 설치여부를 확인합니다.

nvcc -V