[초간단]Ubuntu 20.04 Anaconda, pytorch 설치하기

 

Ubuntu 20.04 딥러닝 환경 구축하기(3) - Cuda toolkit 설치하기

 

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

---

 

딥러닝 환경 구축의 마지막 단계입니다.

 

nvidia driver, cuda까지 설치했다면 이제 Anaconda와 pytorch를 설치해 주면 됩니다.

 

Anaconda를 사용하는 이유는 가상환경 때문이며, 가상환경은 여러 가지 패키지들의 버전 관리를 용이하게 해 줍니다.

 

1. https://www.anaconda.com/download에 들어간 후 화면을 맨 밑까지 내려주면 Anaconda의 Linux install 파일이 있습니다. 64-Bit installer를 설치해 줍니다.

 

 

 

2. Downloads 디렉터리에 Anaconda install 파일이 다운로드되었습니다.

 

 

 

3. 아래 명령어로 Anaconda 설치 파일을 실행해 줍니다.(파일 이름 주의)

bash Anaconda3-2024.02-1-Linux-x86_64.sh

엔터 누르기!

 

 

4. 긴 글들이 뜰 텐데, 스페이스바를 꾹 눌러 밑으로 내려갑니다.

 

 

 

5. yes 입력

 

 

 

6. 또 yes 입력

 

 

 

7. Anaconda 설치가 완료되었습니다. 이제 bash를 수정해주어야 합니다.

 

 

 

8. 아래 명령어로 bash를 열어주고,

sudo gedit ~/.bashrc

 

bash의 맨 밑에 

export PATH=~/anaconda3/bin:~/anaconda3/condabin:$PATH

 

위 코드를 추가해 주고 저장합니다.

source ~/.bashrc

 

를 터미널에 입력해 주면

 

 

9. (base)가 생기면서 anaconda가 정상적으로 작동하는 것을 확인할 수 있습니다.

 

 

 

10. 이제 conda로 새로운 가상환경을 만들고 해당 가상환경에 pytorch를 설치합니다.

 

conda create -n torch python=3.8

 

이름이 'torch'이고 python 버전이 3.8인 가상환경을 만드는 코드입니다. 터미널에 입력해 줍니다.

 

 

 

11. 가상환경이 다 만들어졌다면 해당 가상환경으로 실행해 줍니다.

conda activate torch

 

아래와 같이 가상환경이 실행된 것을 확인할 수 있습니다.

 

 

 

12. 이제 가상환경에 pytorch를 설치해야 합니다.

Previous PyTorch Versions

위 페이지에서 CUDA 버전과 거기에 맞는 pytorch를 설치할 수 있는 명령어를 제공해 주는데, 본인 CUDA 버전에 맞는 pytorch를 선택해 줍니다.

 

저는 CUDA가 11.8이고, pytorch는 최신 버전이 아닌 2.1.0을 설치하고 싶기 때문에 잘 서치 해줍니다.

 

 

13. 해당 명령어를 터미널에서 실행하기.

conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=11.8 -c pytorch -c nvidia

 

위 명령어를 터미널에서 실행해 주세요.

 

 

torch의 용량이 꽤 크기 때문에 시간이 좀 걸립니다.

 

 

14. 위 단계까지 모두 완료되었다면 torch가 정상적으로 작동하는지 확인해봐야 합니다.

 

먼저 터미널에서 python 인터프리터를 실행합니다.

 

import torch
torch.cuda.is_available()

 

위 코드를 순서대로 입력합니다.

 

 

반환값으로 True가 뜬다면 torch가 잘 설치된 것입니다. 끝!