1. 程式人生 > 實用技巧 >win10離線配置GPU版Pytorch

win10離線配置GPU版Pytorch

每次配置環境都令人頭疼,網上教程雖然多,但大多千篇一律,很多不詳細,或者方法不適合自己。

1 確定cudaToolkit版本

開始選單搜尋“計算機管理”→【裝置管理器】→【顯示介面卡】,檢視是否有“NVIDIA”顯示卡,如果沒有,不好意思,請安裝CPU版的Pytorch。

【桌面】→【右鍵】→【Nvidia控制面板】→【系統資訊】→【元件】

2 安裝Cuda執行庫

CudaToolkit官方下載

安裝

建議你按預設安裝,這裡介紹“自定義安裝”

拓展包的儲存位置,這裡預設。

安裝完成後,開啟【控制面板】→【程式和功能】,你會看到如下清單:

驗證。開啟【cmd】,輸入nvcc -V

得到如下介面,表示你安裝成功。

軟體會被強制安裝到C盤,路徑為C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\{你的版本}

3 下載cudnn深度學習加速庫

注:需要先登入cuda才可以下載。

cudnn官方下載

下載後,你會得到一個zip壓縮包,解壓,將裡面檔案拷貝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\{你的版本}

4 安裝PyTorch框架

我嘗試了官方源的命令列線上安裝,失敗了;更換清華大學映象源,也失敗了;最後選擇離線安裝,成功了!

4.1 PiP離線安裝官方源

開啟Pytorch官網

確認版本,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