1. 程式人生 > >【caffe 深度學習】7. win10 caffe gpu版本安裝

【caffe 深度學習】7. win10 caffe gpu版本安裝

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資料夾裡的檔案:


複製到:


選擇替換


至此配置完成。