【caffe】Windows10+Cuda9.0+Cudnn7.0下配置GPU版caffe
阿新 • • 發佈:2018-12-17
第一步:下載所需軟體
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也右鍵生成一下。
搞定!,生成之後會得到下面的一個資料夾!