win10下最好用的pytorch安裝教程

廢話不多說我們直接進入正題。

1、首先確認你有無anaconda，有則跳過這一步，沒有看我這篇博客安裝anaconda，鏈接：https://blog.csdn.net/JohnLeeK/article/details/99582355，然後換源這樣可以加快包的下載速度，參考我的這篇博客，鏈接：https://blog.csdn.net/JohnLeeK/article/details/102686424

2、點擊這個鏈接進入pytorch官網，https://pytorch.org/get-started/locally/，如圖我們按照需求來選擇安裝。

我安裝的是1.2.0的版本，點擊圖中紅色方框鏈接，然後我們找對應版本，這裏安裝可選pip的方式還有conda的方式.。如果安裝1.5的版本直接複製圖中的命令即可（安裝1.2.0版本的往後看，跳過1.5版本）：

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

這裏我選擇用conda的方式安裝（圖二紅色方框命令），因爲pip的方式有點慢（選擇圖三紅色方框命令即可），複製粘貼2中第一幅圖紅色方框內容，要注意的是我選的的cuda10.0的版本，這裏需要注意一個事，打開你命令行窗口（win+r），執行這個命令：

nvidia-smi

結果如下

可以看到我我的驅動版本是417.88，只要你的驅動版本大於410，就安裝cuda10.0的版本或者更版本都沒問題

3、我們先在conda環境中創建一個pytorch的虛擬環境（命令行模式，cmd，快捷鍵win+r）：

conda create -n pytorch python=3.7

然後激活

activate pytorch

安裝pytorch

然後我們執行這個命令（圖二紅色方框複製過來的命令）：

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

4、耐心等待一會兒，完成安裝後我們檢查pytorch是否安裝成功

我們在conda的pytorch虛擬環境中輸入：

python

然後輸入如下代碼

import torch

print(torch.cuda.is_available())

print(torch.__version__)

結果如下，出現這樣的結果就代表的pytorch環境安裝好啦！

可以看到我們安裝的是GPU版本，然後版本爲1.2.0。