小編今天已經在 Windows 10 x64 上的 Anaconda3 5.1.0 上裝了 PyTorch 0.4.0 CPU 版,感覺很好用!現在把安裝方法安利給您們。
小編推薦
使用 conda 命令前,先用以下命令更新 conda 和 python。
conda update conda
conda update python
使用 pip 命令前,先用以下命令更新 pip。
python -m pip install --upgrade pip
為了更快的下載,推薦科學上網~
Windows 安裝命令
Windows x64 + Python 3.5/3.6 + 無 CUDA
conda install pytorch-cpu -c pytorch
pip3 install torchvision
Windows x64 + Python 3.5/3.6 + CUDA 9.1
conda install pytorch cuda91 -c pytorch
pip3 install torchvision
Windows x64 + Python 3.5/3.6 + CUDA 9.0
conda install pytorch cuda90 -c pytorch
pip3 install torchvision
Windows x64 + Python 3.5/3.6 + CUDA 8.0
conda install pytorch -c pytorch
pip3 install torchvision
以上命令中,如果 pip3 不識別,您還可以使用以下命令安裝 torchvision:
python -m pip install --upgrade torchvision
macOS 系統安裝
macOS + Python 2.7/3.5/3.6 + 無 CUDA
conda install pytorch torchvision -c pytorch
Linux 系統安裝
Linux + Python 2.7/3.5/3.6 + 無 CUDA
conda install pytorch-cpu torchvision -c pytorch
Linux + Python 2.7/3.5/3.6 + CUDA 9.1
conda install pytorch torchvision cuda91 -c pytorch
Linux + Python 2.7/3.5/3.6 + CUDA 9.0
conda install pytorch torchvision cuda90 -c pytorch
Linux + Python 2.7/3.5/3.6 + CUDA 8.0
conda install pytorch torchvision -c pytorch
點擊「閱讀原文」訪問 PyTorch 安裝門戶(選擇安裝選項需要科學上網)
↓↓↓