Ubuntu 20.04 RTX 4090 딥러닝 환경 구축

1. APT 업데이트 및 common 드라이버 설치

sudo apt-get update
sudo apt install -y ubuntu-drivers-common

 2. 드라이버 확인

sudo ubuntu-drivers devices

3. nvidia 525 선택

sudo apt install nvidia-driver-525
sudo reboot

4. nvidia 드라이버 잘 설치되었는지 확인

nvidia-smi

5. cuda 11.8 설치

https://developer.nvidia.com/cuda-11-8-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=20.04&target_type=deb_local 

CUDA Toolkit 11.8 Downloads

해당 사이트에서 다음과 같이 선택 후 나오는 명령어 순서대로 입력

 

6. cuda 경로 설정

gedit ~/.bashrc

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

 

7. cuDNN 설치

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

cuDNN Archive

아래 사이트에서 CUDA 11.X 버전에 cuDNN tar 파일 선택 후 압축해제

tar -xvf cudnn-linux-x86_64-8.8.1.3_cuda11-archive.tar.xz

압축 풀린 파일을 cp후 권한 설정

 

sudo cp cudnn-linux-x86_64-8.8.1.3_cuda11-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp cudnn-linux-x86_64-8.8.1.3_cuda11-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