【caffe 深度學習】7. win10 caffe gpu版本安裝
阿新 • • 發佈:2019-02-19
1.準備好NVIDIA的顯示卡,下載安裝CUDA 我這裡是1060的顯示卡
下載地址:
安裝好之後把CUDA安裝目錄下的bin和lib\x64新增到Path環境變數中
最開始我裝的是cuda9.1 裝好之後編譯出錯,改成了cuda8.0加cudnn5.1 成功。
2.cuDNN下載
解壓壓縮包,把壓縮包中bin,include,lib中的檔案分別拷貝到C:\Program Files\NVIDIA GPUComputing Toolkit\CUDA\v9.1目錄下對應目錄中。
3.修改CommonSettings.props檔案
<CpuOnlyBuild>false</CpuOnlyBuild>
<UseCuDNN> true</UseCuDNN>
<CudaVersion>8.0</CudaVersion>
<PythonSupport>false</PythonSupport>
4.開啟Caffe.sln
先編譯libcaffe
點生成
再編譯整個工程 ,完成。
5.mnist手寫數字識別Gpu編譯
首先改下之前寫好的檔案:執行方式改成gpu
再執行train.bat 可以看到明顯速度加快很多。
6.編譯gpu版本的pycaffe,開啟caffewindows/windows目錄下commonsetting檔案
修改為:
(先編輯gpu版本不帶python的caffe,再編譯帶python的caffe不容易出錯)
配置屬性:
然後生成解決方案:
編譯成功:
把此目錄下的caffe資料夾:
複製到:
然後複製caffe資料夾裡的檔案:
複製到:
選擇替換
至此配置完成。