개발/Python

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

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

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
반응형