1. 程式人生 > >win10系統python下實現dlib的GPU加速

win10系統python下實現dlib的GPU加速

因此,需要重新配置環境,實現GPU加速,以提高實效性

1.先解除安裝dlib, pip uninstall dlib

2.安裝VS2015 update3:http://download.microsoft.com/download/8/4/3/843ec655-1b67-46c3-a7a4-10a1159cfa84/vs2015.3.ent_enu.iso

SHA1:40EA340070E3684935689E60D8B7669D519D49D4

3.配置GPU環境:

安裝時選擇自定義,這樣可以選擇安裝資料夾

驅動建議都選了,因為如果驅動版本不對,後面會報錯,還得重新安裝

要下載對應的版本:

下載完成後,將這個壓縮包裡的所有檔案放到CUDA10安裝目錄相應資料夾下即可。

接下來,設定環境變數

可以看到系統中多了

還需要新增:

另外,系統變數 PATH 的末尾新增:

配置完成後,驗證配置是否成功:

cmd進入到D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\demo_suite

執行:bandwidthTest.exe

執行:deviceQuery.exe

結果都顯示pass,就是配置成功。

解壓後cmd進入到dlib安裝資料夾,執行:

安裝結束:

python裡重新呼叫dlib,即可實現GPU加速。