win10 安裝pytorch gpu 及 解決報錯"OSError: [WinError 126] 找不到指定的模塊"
本機參數
GPU：Quadro P620
CUDA：8.0
CUDNN: 11.0.2
python: 3.7
pytorch: 1.7.0
python
安裝CUDA 和 CUDNN
參考網上資源
一、查看顯卡是否支持CUDA及支持的版本。here
二、查看CUDA以及Cudnn版本對應關係。here
三、下載相應版本的CUDA和CUDNN並安裝。here
redis
安裝pytorch
一、官網下載
根據CUDA選擇合適的版本。
建議選擇pip安裝，用conda安裝實在太慢了。 
測試
二、而後把下面的命令複製到anaconda prompt 控制檯。而後等下載安裝完就行了。spa
這個包差很少有2G，能夠參考網上使用鏡像源。不過我就這樣用pip安裝的時候也仍是挺快了。用conda安裝就真是龜速同樣了。
.net
三、測試安裝是否成功。命令行
爲了確保PyTorch是否安裝成功，咱們須要運行簡單的樣例代碼測試，例如打印出隨機生成的張量矩陣，以及gpu是否可使用。debug
首先在命令行輸入python，進入python的解釋器，依次輸入如下命令，每輸入一句回車3d
import torch x = torch.rand(5,3) print(x)
輸出的內容應該相似於如下：code
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]])
若是OK，則說明pytorch安裝基本成功。blog
再輸入
torch.cuda.is_available()
若是輸出「True」，則說明GPU驅動和CUDA能夠支持pytorch的加速計算！
恭喜安裝成功！
報錯解決
今天安裝的時候，照着上面的步驟下來，剛開始仍是報錯了。so sad
OSError: [WinError 126] 找不到指定的模塊。 Error loading"C:\Users\anaconda3\envs\pytorch\lib\site-packages\torch\lib\asmjit.dll"or one of its dependencies.
可是經過查看文件夾，會發現其實這個文件時存在的。就是不知爲何python沒有讀取到。
網上有幾種辦法。好比下降pytorch 版本的。可是我不知道哪些版本適合，一個個試太麻煩，並且怕還要改CUDA版本，嫌麻煩，就沒試。
還有改初始化代碼的。就是報錯的那個文件代碼，修改了路徑，可是也不行。而後一直找不到緣由，拖了一天沒搞定。
最後今晚看到一篇文章。安裝了Microsoft Visual C++ Redistributable，而後就神奇的能夠了。不明覺厲！
雖然不知道是爲何會這樣，可是好在問題解決了，又能夠快樂debug了！搞定收工！