[Linux] 다른 버전의 CUDA Install

Ubuntu 18.04 기준

 

CUDA를 설치하기 전 다음을 진행한다.

sudo rm /etc/apt/sources.list.d/cuda*
sudo apt remove --autoremove nvidia-*
sudo apt remove --autoremove nvidia-cuda-toolkit

sudo apt update

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

 

정상적으로 진행이 되었다면 원하는 CUDA 버전을 아래에서 다운받는다.

 

CUDA Toolkit Archive

 

Platform 선택은 다음과 같이 자신의 OS에 맞게 지정해준다.

위에 있는 Platform 선택을 완료하면 아래에 install하는 command가 친절하게 나온다. 그러면 복붙해서 설치하면 된다.

install 할 때 CUDA Samples, CUDA Demo Suite, CUDA Documentation은 필요없으니 체크 해제한다.

 

만약 driver관련 에러가 뜬다면 "sudo apt install nvidia-driver-" 까지만 입력하고 Tab을 눌러 필요한 버전을 뒤에 입력하여 install한 후 CUDA를 설치할 때 driver는 체크 해제한다. 만약 driver 설치가 안되면 그냥 driver 체크 해제하고 CUDA Toolkit만 체크하고 설치해도 되었다.

 

설치가 정상적으로 완료되었다면 다음과 같이 PATH를 다시 설정해주면 끝이다.

export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}

 

+ Ubuntu 20.04는 cuda 10.2가 지원이 안되니 11.3 버전에 맞추어 install 하면 된다.

+ nvidia driver도 버전에 맞게 설치되어있어야 torch에서 cuda 사용 가능하다. (아래 링크에서 driver runfile을 다운로드받고 설치하면 latest version을 설치할 수 있다.)

https://www.nvidia.com/download/index.aspx