[Ubuntu] NVIDIA RTX 3090을 위한 CUDA 11 설치 방법
새로운 CUDA 11 다운로드를 위해 기존에 CUDA가 설치되어 있다면 충돌 방지를 위해 삭제해줘야 된다.
CUDA 삭제
1
sudo apt-get --purge -y remove 'cuda*'
CUDA 설치
CUDA 공식 사이트에 접속한 후에 [다운로드] 버튼을 눌러 다음 페이지로 이동한다.
위와 같이 필터링하면 맨 마지막에 명렁어 박스가 나온다.
*CUDA 11.2 버전 코드는 아래와 같다.
1
2
3
4
5
6
7
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.2.2/local_installers/cuda-repo-ubuntu2004-11-2-local_11.2.2-460.32.03-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-2-local_11.2.2-460.32.03-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-2-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
자 그러면 CUDA가 설치 완료 됐는데 제대로 설치 됐는지 확인해보자.
1
2
3
4
# 1
nvidia-smi
# 2
nvcc -V
CUDA 11으로 버전이 확인되지 않는다면?
버전이 제대로 확인되지 않으면 설치에서 문제가 있는 것이 아니라 경로 설정이 안 되어 있기 때문이다.
그래서 다음 명령어들로 경로를 맞게 설정해주면 끝난다.
1
2
3
4
5
6
7
8
9
# 설정 파일 열기
vim ~/.bashrc
# 경로 설정
export PATH=/usr/local/cuda-11/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11/lib64
# bashrc 업데이트
source ~/.bashrc