ubuntu 20.04 에 CUDA 11.0 설치 및 Pytorch 1.7.1 설치

이글은 반복되는 Ubuntu 세팅 및 CUDA 설치가 번거로워 한번에 모아보기 위해 참고용으로 작성하였다.

 

주의, Nvidia 드라이버를 따로 설치하지 말고 GUI 사용자는 1번 항목을 따라하지 말고 개별 1. 항목을 따라야한다.

 

1. GUI 기능 끄기(옵션)

$ sudo systemctl set-default multi-user

 

개별 1. Nouveau 비활성화 및 재부팅[1]

$ sudo bash -c "echo blacklist nouveau > /etc/modprobe.d/blacklist-nvidia-nouveau.conf"
$ sudo bash -c "echo options nouveau modeset=0 >> /etc/modprobe.d/blacklist-nvidia-nouveau.conf"
$ sudo reboot now

 

2. 패키지 추가 및 업데이트, CUDA 설치 그리고 재부팅[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 http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda-repo-ubuntu2004-11-0-local_11.0.2-450.51.05-1_amd64.deb
$ sudo dpkg -i cuda-repo-ubuntu2004-11-0-local_11.0.2-450.51.05-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-ubuntu2004-11-0-local/7fa2af80.pub
$ sudo apt-get update
$ sudo apt-get -y install cuda
$ sudo reboot now

 

 

그림 1. 과 같이 nvidia-smi를 입력해 잘 나오는지 확인한다.

 

그림1. CUDA설치후 CUDA버전 및 GPU가 잘 잡히는지 nvidia-smi로 확인할 수 있다.

 

3. 아나콘다 설치 및 파이토치 설치[3, 4]

# 아나콘다 설치(설치 마지막 물음인 conda init 항목은 yes로 해야한다.)
$ wget https://repo.anaconda.com/archive/Anaconda3-2020.11-Linux-x86_64.sh
$ chmod +x ./Anaconda3-2020.11-Linux-x86_64.sh
$ ./Anaconda3-2020.11-Linux-x86_64.sh
$ source .bashrc

# 만약 아나콘다 설치중 마지막 conda init 물음에 no로 답한 경우만 진행
$ ./anaconda3/bin/conda init
$ source .bashrc

# 파이토치 설치
$ conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

 

 

최종적으로 파이토치에서 GPU사용가능 여부를 확인한다.

$ python
>>> import torch
>>> torch.cuda.is_available()

 

참고문헌

1. How to disable/blacklist Nouveau nvidia driver on Ubuntu 20.04 Focal Fossa Linux, Available Online: linuxconfig.org/how-to-disable-blacklist-nouveau-nvidia-driver-on-ubuntu-20-04-focal-fossa-linux 15 Feb 2021.

2. CUDA Tollkit 11.0 Download, Available Online: developer.nvidia.com/cuda-11.0-download-archive 15 Feb 2021.

3. Anaconda Individual Edition Installer, Available Online: www.anaconda.com/products/individual 15 Feb 2021.

4. Pytorch Get Start Locally, Available Online: pytorch.org/get-started/locally/ 15 Feb 2021.