win10離線配置GPU版Pytorch
阿新 • • 發佈:2020-08-11
每次配置環境都令人頭疼,網上教程雖然多,但大多千篇一律,很多不詳細,或者方法不適合自己。
1 確定cudaToolkit版本
開始選單搜尋“計算機管理”→【裝置管理器】→【顯示介面卡】,檢視是否有“NVIDIA”顯示卡,如果沒有,不好意思,請安裝CPU版的Pytorch。
【桌面】→【右鍵】→【Nvidia控制面板】→【系統資訊】→【元件】
2 安裝Cuda執行庫
安裝
建議你按預設安裝,這裡介紹“自定義安裝”
拓展包的儲存位置,這裡預設。
安裝完成後,開啟【控制面板】→【程式和功能】,你會看到如下清單:
驗證。開啟【cmd】,輸入nvcc -V
軟體會被強制安裝到C盤,路徑為C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\{你的版本}
3 下載cudnn深度學習加速庫
注:需要先登入cuda才可以下載。
下載後,你會得到一個zip壓縮包,解壓,將裡面檔案拷貝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\{你的版本}
4 安裝PyTorch框架
我嘗試了官方源的命令列線上安裝,失敗了;更換清華大學映象源,也失敗了;最後選擇離線安裝,成功了!
4.1 PiP離線安裝官方源
確認版本,Python 3.8,cuda 10.1,torch 1.6, torchvision 0.7。
開啟【Anconda Prompt】,輸入pip install {.whl檔案路徑}
安裝torch,然後安裝torchvision
驗證。開啟切換到python模式,結果下圖,那麼恭喜成功了。
4.2 conda離線安裝清華源
如果用迅雷/IDM下載官方源依然很慢,可以下載清華大學映象的,網址,得到.bz2
的檔案,我們用conda離線安裝,輸入命令conda install --offline {.bz2檔名}
解除安裝pytorch命令pip uninstall torch
conda uninstall torch