리눅스 우분투 20.04 LTS [RTX 3090] CUDA 설치기

준비

• 먼저 설치된 환경을 점검합니다.
  

sudo update-pciids

lspci | grep VGA

• 필요한 패키지들을 전부 설치합니다. 각 명령어는 여러 블로그를 참고하며 대입했습니다.
  

#운영체제 패키지

sudo apt update && sudo apt upgrade -y

# 패키지 업데이트 및 필요 패키지 설치

sudo apt install openssh-server
sudo apt install net-tools

# 코드 빌드 및 컴파일 시 필요한 패키지 설치

sudo apt-get install build-essential
sudo apt-get install manpages-dev

• 위 코드를 잘 작성했다면 gcc가 잘 설치 됐는지 확인해봐야 한다.
  

gcc --version

• 이제 자신의 디바이스와 호환되는 nvidia 드라이버를 설치해야한다. 설치 가능한 버젼은 아래 명령어로 확인 가능하다.

ubuntu-drivers devices

nvidia-driver 설치

• nvidia-driver를 설치합니다. 저는 아래 명령어로 설치했습니다.
  

sudo apt install nvidia-driver-515

• 이제 제대로 설치 됐는지 확인하기 위해 아래 코드를 입력합니다.
  

nvidia-smi

조금 주의해야 할 것은 위에 적힌 CUDA Version 은 현재 드라이브 버젼에 딱! 맞는 쿠다 버젼이라고 합니다… 절대로 저게 정답이 아니고 현재 설치된 쿠다 버젼이 저 버젼이 아니예요!… 이걸로 삽질을 엄청 했답니다.. 하하… 아래 사이트를 표를 보고 몇 버전 위의 버전으로만 다운 받으시면 됩니다. https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#id4

• 재부팅 후 와주세요.
  
  
  

쿠다 툴킷 설치

툴킷 다운에 앞서 먼저 파이토치와 텐서플로우에서 지원되는 버젼을 확인 후 설치해야 삽질을 안합니다!…

텐서플로우 버젼 확인 : https://www.tensorflow.org/install/source#gpu
파이토치 버젼 확인 : https://pytorch.org/get-started/locally/

저 같은 경우는 CUDA 11.8로 진행하였습니다.
아래는 11.8 다운 경로이며 상황에 맞는 버젼 툴킷 다운 사이트로 들어가세요!
https://developer.nvidia.com/cuda-11-8-0-download-archive
사양을 선택한 뒤

아래 코드로 다운 받았습니다!

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


sudo sh cuda_11.8.0_520.61.05_linux.run

설치 후 환경 설정을 해주셔야 하는데 먼저 bashrc 파일을 연 후 아래에 코드들을 마지막 줄에 넣어주세요

nano ~/.bashrc

각자 폴더 명이 조금씩 다를 겁니다!..

export PATH=/usr/local/cuda-11.8/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-11.8

cuDNN 설치

먼저 사양을 점검하고 오세요!
https://www.tensorflow.org/install/source#gpu

저의 경우 GPU에서 CUDA 11.8를 사용하는 tensorflow-2.12.0 버젼을 사용할 예정으로 cuDNN은 8.6을 설치하겠습니다.
gcc의 경우 조금 차이 있어도 괜찬은거 같아요 제가 지금 gcc 9.4.0 사용중인데 9.3.1로 권장이지만 정상 작동 확인했습니다.

들어 가셔서 알맞는 버젼의 cuDNN을 설치해 주세요!
https://developer.nvidia.com/rdp/cudnn-archive

저의 경우 11.8 버젼이므로 11.x 버전의 8.6 버전 cuDNN을 tar로 설치 받았습니다.
이제 다운 받은 집 파일을 풀어주세요! 터마널로 진행했지만 이해 쉽게 gui로 보여드릴게요!

안에 들어가시면

이 두 폴더가 있을 겁니다!

이제 저 두 폴더의 파일을 다른 경로로 복사해서 옮길 겁니다! /usr/local 폴더에 가보시면 저희가 설치한 cuda-11.8 폴더가 있을거예요
아래 명령어를 해당 압축푼 cuDNN 폴더 안에서 진행해주세요! 경로명은 상황에 따라 달라질 수 있습니다.

sudo cp include/cudnn* /usr/local/cuda-11.8/include

sudo cp lib/libcudnn* /usr/local/cuda-11.8/lib64

이제 끝!입니다. 고생하셨어요!

각 버젼들에 따라 설치 가능한 파이토치와 텐서플로우, 파이썬 버젼이 정해져 있어요 아래는 저에게 알맞는 버젼 리스트입니다.

텐서플로우 설치

pip3 install tensorflow==2.12.0

파이토치 설치

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

사용가능한 파이썬 버젼

3.8 - 3.11

Written by 유찬영

youngchannel4u@gmail.com