ubuntu20.04 cpu only pytorch 설치하기. GPU없이 파이토치 설치완료

지난번 포스팅은 삽질한 것 같다ㅠㅠ gpu가 없는데 cuda 설치하고 있었음...

CPU only 로 하면 다음과 같은 코드를 치면 된다.

1) 아나콘다로 설치할 경우. conda

conda install pytorch torchvision cpuonly -c pytorch

 

2) pip 로 설치할 경우

pip3 install torch==1.5.1+cpu torchvision==0.6.1+cpu -f https://download.pytorch.org/whl/torch_stable.html

 

나는 저 중에서 1) conda 로 설치했다

conda를 통한 pytorch 설치 cpuonly 로

다음과 같이 뜨면 y 누르고 enter

 

설치가 완료 되었다.

python3

import torch

print(torch.rand(6, 3))

torch.cuda.is_available()

로 쳐서 다음과 같이 확인해 볼 수 있다. 확인하는 겸사겸사 가상환경도 설치해주었다.

conda create -n newenv python=3.8 jupyter notebook numpy scikit-learn matplotlib

 

conda env list

를 보면 새로운 가상환경이 나오는 것을 볼 수 있음...

cpu 라서 학습 속도는 느릴 거라고 한다:)

일단 설치 완료!