由於項目需要,所以在自己本子上配置了一下windows下GPU版本的caffe;
硬件: win10 ; gtx1070獨顯(計算能力6.1);
安裝軟件: cudnn-8.0-windows10-x64-v5.1 ; cuda_8.0.61_win10 ; NugetPackages.zip ; caffe-master;
可以自己官網下載(我也提供了百度雲:鏈接:https://pan.baidu.com/s/1miDu1qo 密碼:w7ja)
參考鏈接: https://www.cnblogs.com/king-lps/p/6553378.html(主要按照這篇和下面兩篇來配置的)
一個word版的配置教程:鏈接:https://pan.baidu.com/s/1hsEiDEG 密碼:d29u
http://www.cnblogs.com/zf-blog/p/6139044.html
http://blog.csdn.net/tina_ttl/article/details/51722983
https://www.2cto.com/kf/201612/575865.html
具體的詳細步驟我就先不寫了,之前一直在ubuntu上用的caffe,這次配置很輕松,我只說一下幾點注意的地方:
(1)安裝cuda時,注意 要取消勾選安裝driver compoent選項;
(2)cuda安裝好后會自動添加到系統變量中,而安裝完cuda后最好驗證一下(nvcc -V和運行deviceQuery.exe,兩者最好都跑一下,后者我運行時出現閃退情況,嘗試解決失敗,但也沒有影響最后的安裝);
(3)下載解壓完cudnn后,要將解壓后得到的三個文件夾合並到cuda相應的文件夾中就OK了;
(4)下載caffe時主要要下載windows版本的caffe,就是文件夾中要有windows這個文件夾;
(5)編譯的時候不需要全部編譯,編譯順序為libcaffe -> caffe ,如果需要再編譯python或者matlab接口;
然后就OK了,試試最簡單的mnist就行了;