win10 安裝pytorch gpu 及 解決報錯"OSError: [WinError 126] 找不到指定的模塊"
- 本機引數
- 安裝CUDA 和 CUDNN
- 安裝pytorch
- 報錯解決
本機引數
GPU：Quadro P620
CUDA：8.0
CUDNN: 11.0.2
python: 3.7
pytorch: 1.7.0
安裝CUDA 和 CUDNN
參考網上資源
1、查看顯卡是否支持CUDA及支持的版本，here
2、查看CUDA以及Cudnn版本對應關系，here
3、下載相應版本的CUDA和CUDNN并安裝，here
安裝pytorch
1、官網下載
根據CUDA選擇合適的版本，
建議選擇pip安裝，用conda安裝實在太慢了， 
2、然后把下面的命令復制到anaconda prompt 控制臺，然后等下載安裝完就好了，
這個包差不多有2G，可以參考網上使用鏡像源，不過我就這樣用pip安裝的時候也還是挺快了，用conda安裝就真是龜速一樣了，
3、測驗安裝是否成功，
為了確保PyTorch是否安裝成功，我們需要運行簡單的樣例代碼測驗，例如列印出隨機生成的張量矩陣，以及gpu是否可以使用，
首先在命令列輸入python，進入python的解釋器，依次輸入以下命令，每輸入一句回車
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]])
如果OK，則說明pytorch安裝基本成功，
再輸入
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了！搞定收工！
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/206957.html
標籤：其他
下一篇：自動化測驗--自動化測驗報告