安裝win10
安裝microsoft visual studio 2015的運行環境(kb4020481),這是安裝pytorch的必要條件
安裝anaconda3
加入清華源 #因爲官方源速度不錯,這一步能夠不作
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
安裝pytorch
安裝依賴包
conda install numpy mkl cffi
conda install pytorch-cpu torchvision-cpu #不能使用CUDA,用CPU作運算
conda install pytorch torchvision #使用CUDA,用GPU作運算
驗證PyTorch 安裝
在命令行運行python,輸入如下代碼:
from future import print_function
import torch
x = torch.rand(5, 3)
print(x)python
輸出結果相似於: 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]])