[GPU] NVIDIA 드라이버 & CUDA Toolkit 설치하기 / 2023.06.15

GPU를 사용하기 위해서 필요한 것들을 설치해보자!

Step 1. NVIDIA 드라이버 설치

1-1. 설치가능한 드라이버 버전 확인

ubuntu-drivers devices

 

1-2. 드라이버 설치 (470 선택)

sudo apt install nvidia-driver-470

 

1-3. 재부팅

sudo reboot

 

1-4. 설치한 드라이버 정보 확인

nvidia-smi

 

Step 2. CUDA Toolkit 설치하기

2-1. CUDA Toolkit 선택

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

CUDA Toolkit 11.7 Downloads

본인의 운영체제에 맞게 선택해주면

아래에서 명령어를 볼 수 있음

2-2. CUDA Toolkit 설치용 파일 다운로드

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

 

2-3. 설치 파일 실행

sudo sh cuda_12.1.1_530.30.02_linux.run

 

Continue -> Enter

accept -> Enter

Space눌러서 Driver 체크박스 해제 후, Install -> Enter

설치 완료!!

2-4. 환경 변수 추가

sudo sh -c "echo 'export PATH=$PATH:/usr/local/cuda-12.1/bin' >> /etc/profile"
sudo sh -c "echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-12.1/lib64' >> /etc/profile"
sudo sh -c "echo 'export CUDADIR=/usr/local/cuda-12.1' >> /etc/profile"
source /etc/profile

 

2-5. 설치된 CUDA Toolkit 확인

nvcc --version