[우분투] 파이토치 설치 쉽게하기

파이토치 설치하는 방법입니다.
딥러닝을 구현하려면 텐서플로우랑 파이토치가 필요한데 텐서플로우는 이미 conda나 pip으로 간단하게 설치가 가능한데 파이토치는 그렇지 않습니다. 파이토치 홈페이지에서 설치파일을 받아야 합니다.

설치파일 받기

파이토치는 홈페이지에 가서 개발환경 버전에 맞는 설치파일을 가져와야 합니다. 기존에 있던 www.pytorch.org 는 보안과 안정성 때문에 좀 찝찝한 부분이 있어서 추천드리지 않고 파이토치 한국 사용자 모임 홈페이지에서 pytorch 설치를 진행하는 것을 추천합니다.아래 홈페이지로 가면 됩니다.
https://pytorch.kr/get-started/locally/

파이토치 한국 사용자 모임 (PyTorch Korea User Group)

글씨 캡처는 안 되게 해놔서 그림이 명확하지 않네요 ㅜ

아무튼 들어가서 해당 버전에 맞는 것을 지정하면 아래에 명령어가 나옵니다. 그 명령어를 터미널에서 작성하면 설치가 시작됩니다. cuda 11.6이 최신이니 개발환경의 cuda 버전이 더 높다면 cuda 11.6버전으로 설치파일을 받도록 합니다.

cuda 버전 확인을 할 줄 모른다면 아래 내용을 확인하시기 바랍니다.

 

 

 

 

 

cuda 확인

리눅스에서 cuda 버전을 확인하고 싶다면 터미널에서 다음을 입력합니다.

nvidia-smi

저는 cuda version이 11.7이네요 이런 식으로 gpu인식을 확인한 후 자신에 맞는 설치파일을 받으시기 바랍니다.

파이토치 설치하기

받은 명령어를 터미널에 입력해서 설치를 진행합니다.
리눅스 터미널이라면 conda로 되어있지 않아서 conda 명령어가 먹히지 않습니다. conda activate를 한 상태에서 해야합니다.
conda activate를 하면 앞에 (base)가 나타납니다. 이 상태에서 설치를 진행합니다.

 

Solving environment: failed with initial frozen solve.

혹시 soving environment에서 멈춰있다면 conda가 최신버전이 아니어서 생기는 문제입니다. 이런 경우 ctrl+c로 빠져나온 후 conda를 최신버전으로 업데이트 합니다. 다음 명령어로 업데이트를 진행합니다.

conda update --all

 

설치확인

터미널에서 python을 들어가 import torch를 해봅니다.
잘 설치되었다면 오류없이 명령이 실행될 것입니다.