[설치] Ubuntu에 Pytorch -GPU 버전 설치하기

 

 

업무용 컴퓨터로 우분투를 쓰고 있습니다. 우분투 22.04 버전이구요,

항상 개발 업무를 하는 데 있어서 가장 어려운 점은 바로 설치!! install 입니다~!

 

설치할 때 가장 오류가 많이 나기 때문인데요,

이번에 가상 환경을 달리하여 pytorch를 다시 설치하면서 또 버벅 대길래 정리할겸 블로그에 글을 남깁니다.

 

 

0. 지금 현재 상황!

gpu가 깔려있는 ubuntu 컴퓨터를 쓰고 있고,
파이썬 가상환경을 만들어서 pytorch를 까는데, gpu 를 사용하는 버전으로 깔고 싶다.

 

 

1. 문제 해결법은 간단... 한 줄 알았다..!

그냥 이 명령어만 입력하면 되기 때문이죠.

conda install -c conda-forge pytorch-gpu

 

이 명령어를 내가 만든 가상환경으로 conda activation 하여서 터미널창에 입력하면 시간이 좀 걸리면서 됩니다.

그냥 conda install pytorch로 하면 안 되고, 꼭 gpu 버전으로 (위 명령어) 설치해야만 gpu를 사용하는 pytorch를 설치할 수 있습니다.

 

라고 생각하려고 했는데, 

import torch 를 했을 때 오류가 났다.

 

그래서 그냥 공식 홈페이지에서 지침에 따라 다운로드 하자고 판단!!

 

2. 문제 해결법은 진짜 간단했다,, 역시 공식 홈페이지에서..!!

pytorch.org/get-started/locally/

홈페이지로 들어가니 아래와 같은 그림이 나왔다.

 

나한테 맞는 속성들을 다 클릭하고, (Linux, Conda, Python, CUDA 11.7)

Run this Command: 뒤에 있는 저거를 복사해서 터미널창에 붙여넣기를 해주면 설치 끄읕~~~!

 

 

 

오!! 진짜 그러니까 import torch가 되었다.

 

import torch

print(torch.__version__)

 

2.0.1 이라고 뜸..!!