Ubuntu 22.04 x RTX 4090 CUDA 및 PyTorch 설치

23년 07월에 3090에서 4090 기준으로 재작성되었습니다

GPU 확인

하드웨어 정보를 확인한다.

lspci | grep -i nvidia

여기 보면 난 분명히 4090인데 어째서 4090 이름은 안 나오고 엉뚱한 2684가 나왔다. 3090일 때는 3090 이름으로 잘 나왔었는디...

검색해보니 40XX 시리즈의 GPU ID가 나오는 것 같다. 해당 주소 본문에 NVIDIA_DEV.2684 = “NVIDIA GeForce RTX 4090” 라고 써있었다... 흠 맞나보군 !

GCC Complier

sudo apt install build-essential
gcc --version

설치가 완료되면 gcc --version 으로 설치가 완료되었는지 확인한다.

Ubuntu Driver

sudo apt install software-properties-common -y
sudo add-apt-repository ppa:graphics-drivers/ppa -y
sudo apt update

ubuntu-drivers devices

# recommended로 자동설치
sudo ubuntu-drivers autoinstall

# 재부팅
sudo reboot now 
nvidia-smi

nvidia-smi를 통해 확인한 쿠다버전은 compatible 한 버전이다

재부팅 이후 설치 확인

CUDA

Pytorch 확인

쿠다 설치하기 전에 Pytorch Installation 페이지에서 # INSTALL PYTORCH 부분에서 지원하는 쿠다 버전을 확인한다.

우분투 22.04 환경은 쿠다 11.7부터 공식 지원하지만, 파이토치는 아직 쿠다 11.6을 지원하기 때문에 11.6으로 설치가 필요하다.
23.07 업데이트
우분투 22.04 환경에서 11.8을 설치하도록 한다! (GPU는 4090 사용 중 )

CUDA Runfile 다운로드

• runfile이 아니어도 설치에는 무관합니다!
  CUDA toolkit 사이트에서 11.8로 다운로드 받는다.

Bash Installer 나온 명령어를 터미널 위에서 실행시켜준다.

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

CUDA Runfile 실행

sudo sh cuda_11.8.0_520.61.05_linux.run

Driver 해제 필수 !

해당 화면이 터미널에 뜨면 Continue > accept > (Driver 해제) Install 의 순서로 진행한다.

CUDA PATH 설정

설치가 완료되면 위와 같이 결과가 뜬다.

PATH includes … LD_LIBRARY_PATH includes … 부분을 확인한다.

터미널에서

1. 본인의 쉘을 확인
   1. 우분투에서 oh-my-zsh 등을 설치하지 않았다면 기본은 bash다.
2. 쉘 설정 파일을 열어준다.
3. 설치 결과에 나왔던 PATH, LD_LIBRARY_PATH를 설정파일에 넣고 저장해준다.
   1. export PATH=”{설치 결과에 나온 PATH 주소}:$PATH”
export LD_LIBRARY_PATH=”{설치 결과에 나온 LD_LIBRARY_PATH 주소}:$LD_LIBRARY_PATH”

# 1. 쉘 확인
# zsh인지 bash인지 확인이 가능
echo ${SHELL} 

# 2. 설정파일 열기
# bashrc or zshrc
vim ~/.zshrc

작성자는 zsh을 사용하므로 본 화면은 실제 실행환경과 다를 수 있습니다.

최종 확인

파이썬을 열고 파이토치를 불러와 gpu 가 잡히는지 확인한다.