1. 程式人生 > >【caffe】Windows10+Cuda9.0+Cudnn7.0下配置GPU版caffe

【caffe】Windows10+Cuda9.0+Cudnn7.0下配置GPU版caffe

第一步:下載所需軟體

1、下載CUDA9.0,也是目前的最新版本。

2、下載CUDNN V7.0。

3、下載Caffe。

 第二步:安裝和配置

參考網上的大多數教程,對軟體進行安裝配置。

1、安裝CUDA9.0,安裝過程很簡單。

2、CUDA9.0安裝完成後,開啟CUDNN檔案,將裡面的bin、lib、include三個資料夾複製到CUDA9.1的安裝路徑中與CUDA9.1本身存在的三個資料夾合併。

3、將caffe檔案存放到一個資料夾,將caffe-master\windows\下的CommonSettings.props.example另存一份命名為CommonSettings.props,開啟caffe-master\windows\Caffe.sln,然後開啟工程中的配置項。

4.修改此處的工程配置。

5.右鍵libcaffe,點選生成。

6.此時會報錯,先僅編譯生成  libcaffe  時會報錯   error: too few arguments in function call,進而導致後面124個錯誤。

雙擊檢視報錯函式   cudnnSetConvolution2dDescriptor,在引數列表末尾補充一個引數 CUDNN_DATA_FLOAT 或CUDNN_DATA_DOUBLE即可。

6.最後再重新生成libcaffe,生成成功之後,再可以把caffe也右鍵生成一下。

搞定!,生成之後會得到下面的一個資料夾!