Win10+Pytorch-0.3.1+Anaconda3-5.1.0+CUDA_9.1.85+cudnn-9.1

之前在VMware虛擬機器上使用Pytorch，結果虛擬機器出於某些原因啟動後黑屏，因此索性配置一個Win10環境下的Pytorch程式設計環境，在此記錄下自己的配置過程。

本文已過期，Pytorch已經推出支援Windows，可以參考這篇文章

需要準備的東西：

• Anaconda包管理工具，直接百度下，選擇對應系統的版本，如我的是Windows64位，想用Python3.6，則選擇64-Bit Graphical Installer。下載完後安裝即可，安裝過程中注意勾選配置環境變數即可，方便後續使用conda操作
• CUDA工具庫官網下載連結，選擇對應系統及對應配置，如我的是Win10+86_64位+local版本（也可以嘗試下network版，如果網速沒問題不會出現斷點）
• cudnn庫 官網下載連結 ，是CUDA用來加速深度神經網路的一個庫，在官網下載cudnn需要註冊NVIDIA的賬號並且填寫一些調查問卷題（很快的。。。不能急這個）
• VS2017（可選）這個在安裝CUDA時會提示如果沒安裝的話可能部分CUDA函式沒法使用（具體是什麼我也不清楚，所以我安裝了），百度下載即可，微軟對中國使用者的支援還是可以的。

Pytorch的安裝

最關鍵的來了，下載安裝過程如下：（參考自知乎大神蒲嘉宸 ）
使用Anoconda命令列（安裝的是基於CUDA9版的Pytorch，必須有對應N卡和CUDA驅動）：

conda install -c peterjc123 pytorch cuda90

結果發現網路老是斷點，沒有耐心等待成功的我只好選擇另一種方式，通過百度雲下載大神做好的包，我選用的是pytorch-0.3.1-py36_cuda90_cudnn7he774522_2.tar.bz2，關於引數的說明參考蒲嘉宸知乎連結

總之，最後下好了(感謝百度雲的免費會員體驗加速，賊快)，命令列下進入Pytorch包下載後所在目錄，執行：

conda install numpy mkl cffi
conda install pytorch-0.3.1-py36_cuda90_cudnn7he774522_2.tar.bz2
 

等待片刻，完成安裝（注：若過程中出現錯誤CondaError: EOFError('Compressed file ended before the end-of-stream marker was reached',)，說明之前下載的Pytorch包不完整，存在快取，因此必須到相應路徑刪除掉再執行上述命令。路徑為Anaconda下pkgs目錄，如我的：
C:\Users\zsn\Anaconda3\pkgs\pytorch-0.3.1-py36_cuda90_cudnn7he774522_2.tar.bz2）

開啟python，測試安裝是否成功：

import torch
x = torch.Tensor([1.0])
xx = x.cuda()
print(xx)

# CUDNN TEST
from torch.backends import cudnn
print(cudnn.is_acceptable(xx))

注：引用自蒲嘉宸知乎連結 “安裝之後，也千萬要注意，要在主程式碼的最外層包上

if __name__ == '__main__':

這個判斷，可以參照我昨天文章中的例子，因為PyTorch的多執行緒庫在Windows下工作還不正常”

順便說一下再安裝torchvision，使用anconda即可：

pip install torchvision

最後，祝賀自己成功安裝，感謝一下網上各路大神的部落格資料~