업무용 컴퓨터로 우분투를 쓰고 있습니다. 우분투 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 이라고 뜸..!!
'데이터사이언스 > 파이썬' 카테고리의 다른 글
[개발자일지 #01] 병렬처리 (0) | 2023.05.11 |
---|---|
[파이썬 기초] print 함수 (0) | 2022.12.10 |