개발/Python

PyTorch CPU version install (Windows / Linux) (PIP / Anaconda)

강서버 2021. 6. 18. 19:09
반응형

PyTorch + CPU (Not CUDA)

 

1. PIP

[ Windows ]
1.1 Stable(1.9.0)
pip3 install torch torchvision torchaudio

 

1.2 LTS(1.8.1)
pip3 install torch==1.8.1+cpu torchvision==0.9.1+cpu torchaudio===0.8.1 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html

 

[ Linux ]
1.1 Stable(1.9.0)

pip3 install torch==1.9.0+cpu torchvision==0.10.0+cpu torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

 

1.2 LTS(1.8.1)

pip3 install torch==1.8.1+cpu torchvision==0.9.1+cpu torchaudio==0.8.1 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html

 

 

2. Connda

[ Windows & Linux ]
2.1 Stable(1.9.0)
conda install pytorch torchvision torchaudio cpuonly -c pytorch

 

2.2 LTS(1.8.1)
conda install pytorch torchvision torchaudio cpuonly -c pytorch-lts

 

참고 : https://pytorch.org/get-started/locally/

 

PyTorch

An open source machine learning framework that accelerates the path from research prototyping to production deployment.

pytorch.org

 

 

728x90
반응형

'개발 > Python' 카테고리의 다른 글

Python 3.9 pip run error  (0) 2022.01.11
PyCharm PEP8 setting  (0) 2021.06.18
ImportError: ('You must install pydot (`pip install pydot`) and install graphviz  (0) 2021.06.15
Python runtime measurement  (0) 2021.06.11
Anaconda3 Update  (0) 2021.06.09