[Ubuntu] GPU 설정하기 - Nvidia Driver, CUDA, cuDNN 설치

 

Ubuntu 22.04

RTX 3060

 

호환성과 안정성을 위하여 nvidia driver 580 최신 드라이버 보다는 535 버전을 설치하였다.

 

 

1. nvidia driver 설치
sudo apt install nvidia-driver-535
sudo reboot


# reboot 한 다음 출력내용 확인하기
nvidia-smi

 

 

2. CUDA 12.1 run 파일 설치

 

wget https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_530.30.02_linux.run

 

chmod +x cuda_12.1.1_530.30.02_linux.run
sudo sh cuda_12.1.1_530.30.02_linux.run

 

 

CUDA 12.1 설치 + NVIDIA Driver 535.xx 설치가 한 번에 가능하지만 Driver 는 이미 설치했으므로, 

설치 중 다음 옵션 선택:

Install NVIDIA Driver → No
Install CUDA Toolkit → YES
Install Samples → 원하면 No

 

환경변수 등록
sudo nano ~/.bashrc

export PATH="/usr/local/cuda/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"

source ~/.bashrc

# 명령어 입력 시 제대로 출력되면 성공 -> 내용 확인
nvcc -V

 

 

3. cuDNN 설치 - 라이브러리 설치

 

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

cuDNN Archive

 

cuDNN v8.9.7 for CUDA 12.x
Ubuntu22.04 x86_64 (Deb)

 

다운받은 파일을 압축 풀기
tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda12-archive.tar.xz

sudo cp cudnn-linux-x86_64-8.9.7.29_cuda12-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp cudnn-linux-x86_64-8.9.7.29_cuda12-archive/lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

아래 명령어 입력시 제대로 출력되면 성공

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

 

--------------
#define CUDNN_MAJOR 8
#define CUDNN_MINOR 9
#define CUDNN_PATCHLEVEL 7

--------------

 

 

 

참고 : [개발환경] RTX 4090 Ubuntu 22.04 딥러닝 학습 서버 환경 구축 (1)

[개발환경] RTX 4090 Ubuntu 22.04 딥러닝 학습 서버 환경 구축 (1)