###내 데스크탑 정보###

• CPU : i7-10700
• RAM : 32Gb
• GPU : RTX 3080ti
• Ubuntu 18.04

1. NVidia 드라이버 설치

Software Updater에 들어가면

왼쪽 하단의 Settings..에 들어가

이와 같이 Additional Drivers에 들어가면 다운 받을 수 있다.

난 이미 다운 받았기때문에 Apply Changes 가 뜨지 않지만 뜨게 될 것이다. GPU가 나와 다르다면 그에 맞는 nvidia-driver-4xx 번호에 맞게 다운 받으면 된다.

터미널에서

nvidia-smi

라고 치면 아래와 같이 나오면 성공이다.

상단에 CUDA Version : 11.5라고 쓰여있기 때문에 난 11.5 버전을 다운 받았다.

2. CUDA 11.5 설치

https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=18.04&target_type=runfile_local

위의 사이트에 들어가서 본인 컴퓨터 사양에 맞춰 선택하면 된다.

난 위와 같이 선택했고 Base Installer에 터미널에 입력하라는대로 입력하면 된다.

continue 를 선택하고 accept 입력하고 driver는 이미 설치했기 때문에 driver는 엔터나 스페이스바를 통해 체크를 해제한 후 엔터를 눌러 설치하면 된다.

3. cuDNN 8.3.0 설치

!!!NVIDIA 로그인 필수!!!

NVIDIA에서 회원가입 후 설문조사를 마치면

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

위의 사이트에서 다운 받을 수 있다.

나한테 맞는 deb 파일을 선택 후 다운 받아 압축을 풀어준다.

sudo cp /data/usr/include /usr/local/cuda/include cudnn*.h
sudo cp /data/usr/lib /usr/local/cuda/lib64 libcudnn*
chmod a+r /usr/local/cuda/include /usr/local/cuda/lib64
cat /usr/local/cuda/include/cudnn_version_v8.h | grep CUDNN_MAJOR -A 2

위의 명령어를 통해 파일들 옮겨주고 chmod 해준 후, CUDNN 설치 확인해주면 된다.

위와 같이 뜬다면 설치 성공이다. 해당 파일이 없다고 뜬다면 cudnn_version~으로 나오는 파일명으로 바꿔주면 된다.

4. 환경변수 설정

vi ~/.bashrc

• vi 하단에 PATH 추가

export PATH=/usr/local/cuda-11.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH

esc 키 누른 후 :wq 치고 엔터키 눌러서 저장해준다.

• 수정한것 저장하기

source ~/.bashrc

• CUDA 버전 확인하기

nvcc -V

이제 CUDA, CUDNN 설치는 끝이다.