選擇首選項並運行install
命令。Stable
代表PyTorch 1.1最受測試和支持的版本,它應該適合許多用戶。如果您想要每晚生成的最新的,也可以安裝未經過完全測試和支持的1.1版本。首先請確保已滿足以下先決條件(例如,numpy),具體取決於您的包管理器。Anaconda是推薦的軟體包管理器,因為它安裝了所有依賴項。
在Windows上安裝PyTorch
PyTorch可以在各種Windows發行版上安裝和使用。根據您的系統和計算要求,在Windows上使用PyTorch的體驗可能會因處理時間而異。建議(但不要求)Windows系統具有NVIDIA GPU,以便充分利用PyTorch的CUDA支持。
必備條件
支持Windows分發
以下Windows發行版支持PyTorch:
- Windows 7及更高版本; 建議使用Windows 10或更高版本。
- Windows Server 2008 r2及更高版本
此處的安裝說明通常適用於所有受支持的Windows發行版。顯示的具體示例將在Windows 10 Enterprise電腦上運行。
Python
目前,Windows上的PyTorch僅支持Python 3.x; 不支持Python 2.x,請務必瞭解。由於Windows上沒有默認安裝,因此有以下幾多種方法可以安裝Python:
如果使用Anaconda安裝PyTorch,它將安裝一個沙盒版本的Python,用於運行PyTorch應用程式。如果決定使用Chocolatey,但尚未安裝Chocolatey,請確保以管理員身份運行命令提示符。
對於基於Chocolatey的安裝,請在管理命令提示符中運行以下命令:
choco install python
包管理器
要安裝PyTorch二進位檔,需要使用兩個受支持的軟體包管理器中的其中一個:Anaconda和pip。Anaconda是推薦的包管理器,因為它將在一個沙盒安裝中提供所有PyTorch依賴項,包括Python和pip。
- Anaconda - 要安裝Anaconda,可使用PyTorch 3.x的64位圖形安裝程式。單擊安裝程式鏈接,然後選擇“運行”。Anaconda將下載並將向您顯示安裝程式提示。
- pip - 如果是通過上面推薦方式安裝了Python,則已經安裝了pip。
- Numpy - 如果是通過pip安裝的,需要在安裝PyTorch之前安裝numpy。
# Python 3.x pip3 install numpy
執行安裝
使用Anaconda安裝
要使用Anaconda安裝PyTorch,需要通過Start |打開Anaconda提示符 Anaconda3 | Anaconda提示。
沒有CUDA
要通過Anaconda安裝PyTorch,在沒有支持CUDA的系統或不需要CUDA,請使用以下conda命令。
conda install pytorch-cpu torchvision-cpu -c pytorch
要通過Anaconda安裝PyTorch,並且使用的是CUDA 9.0,請使用以下conda命令:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
## CUDA 8.x
conda install pytorch torchvision cudatoolkit=8.0 -c pytorch
## CUDA 10.0
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
使用PIP安裝
沒有CUDA
要通過pip安裝PyTorch,並且沒有支持CUDA的系統或不需要CUDA,請使用以下命令,具體取決於您的Python版本:
# Python 2.7
pip install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp27-cp27mu-linux_x86_64.whl
pip install torchvision
# if the above command does not work, then you have python 2.7 UCS2, use this command
pip install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp27-cp27m-linux_x86_64.whl
其他版本的安裝:
# Python 3.5
pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision
# Python 3.6
pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision
# Python 3.7
pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp37-cp37m-win_amd64.whl
pip3 install torchvision
CUDA 9.0
要通過pip安裝PyTorch,並且您使用的是CUDA 9.0,請使用以下命令,具體取決於您的Python版本:
# Python 3.5
pip3 install https://download.pytorch.org/whl/cu90/torch-1.0.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision
# Python 3.6
pip3 install https://download.pytorch.org/whl/cu90/torch-1.0.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision
# Python 3.7
pip3 install https://download.pytorch.org/whl/cu90/torch-1.0.1-cp37-cp37m-win_amd64.whl
pip3 install torchvision
# Python 3.x
pip3 install torch torchvision
# Python 2.7
pip install torch torchvision
驗證
為確保正確安裝PyTorch,可以通過運行示例PyTorch代碼來驗證安裝。這裏將構造一個隨機初始化的張量。
從命令行輸入:
C:> python
然後輸入以下代碼:
from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)
得到類似以下的結果:
tensor([[0.3380, 0.3845, 0.3217],
[0.8337, 0.9050, 0.2650],
[0.2979, 0.7141, 0.9069],
[0.1449, 0.1132, 0.1375],
[0.4675, 0.3947, 0.1426]])
此外,要檢查PyTorch是否啟用並可訪問GPU驅動程式和CUDA,請運行以下命令以返回是否啟用了CUDA驅動程式:
import torch
torch.cuda.is_available()