[초간단]Ubuntu 20.04 Cuda toolkit 설치하기

https://mvcv.tistory.com/13

Ubuntu 20.04 딥러닝 환경 구축하기(2) - Nvidia driver 설치(nvidia-smi)

 

이전 글(Nvidia driver 설치하기)에서 이어집니다.

---

 

Ubuntu 20.04를 설치하고 Nvidia driver도 설치했다면, 이제 Cuda toolkit을 설치할 차례입니다.

 

Cuda toolkit 설치는 굉장히 간단합니다.

 

저는 cuda 11.8 버전을 설치했습니다. 아직 최신버전이 불안해서 11.8을 선택했습니다.

 

1. https://developer.nvidia.com/cuda-11-8-0-download-archive에 들어가 줍니다. Cuda 11.8 설치 파일을 다운로드하는 홈페이지입니다.

본인의 환경에 맞춰 클릭해 줍니다. 저는 위처럼 해주었습니다.

 

 

2. 설정을 마치면 아래 화면에 보이는 명령어들이 밑에 뜰 겁니다. 이 명령어들을 차례대로 터미널에 입력해 줍니다.

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.8.0/local_installers/cuda-repo-ubuntu2004-11-8-local_11.8.0-520.61.05-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-8-local_11.8.0-520.61.05-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-11-8-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda

 

파일 용량이 크기 때문에 설치하는데 시간이 좀 걸립니다.

 

3. 위 과정이 다 완료됐다면, 이제 cuda의 환경변수를 bash에 추가해주어야 합니다.

sudo gedit ~/.bashrc

 

위 명령어로 bash 파일을 열어주고,

export PATH="/usr/local/cuda-11.8/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH"

 

위 두줄을 bash 파일 맨 밑에 추가해 줍니다.

source ~/.bashrc

 

위 명령어로 수정된 bash를 적용합니다.

 

4. 마지막으로 터미널에 'nvcc -V' 명령어를 입력해 준다면?

위와 같이 Cuda 11.8이 잘 설치된 것을 확인할 수 있습니다. 끝!