[Install] 내 PC에 맞는 CUDA 설치하기

내 컴퓨터에 맞는 CUDA를 설치해보자

정말 간단하다

nvidia driver가 설치되어 있던 아니면 전부 적용 가능하다

Nvidia driver version 확인

이 명령어를 입력해주자
가운데 출력되는 Driver Version 값이 현재 PC에 설치된 nvidia driver 버전이다
나의 경우에는 nvidia-driver-515가 설치되어 있다
오른쪽에 뜨는 CUDA Version은 내 PC에 설치된 드라이버에서 최대로 설치할 수 있는 CUDA 버전이다
근데 보통은 맞게 설치하기 때문에 거기 적혀있는 대로 설치하자

$ nvidia-smi

내 PC 환경

GPU: Nvidia RTX 3070
드라이버: nvidia-driver-515
CUDA: cuda11.7

위와 같이 설치해 줄 예정이다

설치방법

“CUDA Toolkit Archive” 여기에 들어가서 해당되는 것을 선택하자
나의 경우 제일 위에 있는 CUDA 11.7을 선택하였다

Installer Type은 아래 두 가지 방법을 추천한다

• deb (local)
• runfile (local)

나는 runfile로 설치할 예정이다
아래와 같이 선택을 해주면,

아래와 같이 어떻게 설치하면 되는지 명령어를 다 알려준다 !

아래 명령어를 따라 설치해주자

$ wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
$ sudo sh cuda_11.7.0_515.43.04_linux.run

설치 내용

처음에 설치하면 EURA 같은게 뜨는데 라이센스 약관에 동의한다는 내용이다
accept를 입력하자

그 후에 내가 설치해야하는 것에 X표시가 마킹되어 있는데,
이 때 그래픽 드라이버가 없으면 그래픽 드라이버에도 표시해주고, 이미 그게 설치가 되어 있으면 체크 해제를 하자 !

환경변수에 적용!

CUDA를 환경변수에 적용해주자

$ gedit ~/.bashrc

아래 줄을 입력해준다
여기서 cuda-11.7이라고 적혀있는데, 본인이 설치한 버전에 맞게 넣어준다
그 경로는 다음과 같다 cd /usr/local/ 경로에서 ls 하면 어떤 것이 설치되어있는 지 알 수 있다

# For CUDA
export PATH="/usr/local/cuda-11.7/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.7/lib64:$LD_LIBRARY_PATH"

적용!

$ source ~/.bashrc

끝!