win7/win8+VS2013+caffe配置並測試Mnist

本來只是想寫win7的,但是經證實win8也是可以的,過程都一樣

配置階段

配置參考:

http://blog.csdn.net/tjusxh/article/details/48463409
http://blog.csdn.net/happynear/article/details/45372231
非常感謝happynear和tjusxh的分享!

一 準備

  1. VS2013

  2. CUDA7.5
    記住一定要先裝VS再裝CUDA!

  3. 已經帶好第三方庫的caffe,直接下載http://pan.baidu.com/s/1TBtvC

二 編譯

  1. 將3rdparty/bin文件夾添加到環境變量的PATH中,這樣才能讓程序找到這些第三方庫的dll 很重要!!

  2. 打開buildVS2013文件夾中的 MainBuilder.sln文件,如果無法加載,表明CUDA版本不同。進入 buildVS2013目錄,每個文件夾中的 .vcxproj (好像是有7個)用editplus打開,把所有的CUDA版本換成你自己的,如7.5換成7.0。

  3. 點擊工具欄的綠色三角調試。第一次好像挺久的,半個小時左右。成功後將在根目錄的bin中生成caffe.exe。雙擊,小黑屏一閃而過(沒有example不會開始訓練的)。好像默認是調試第一個,要是調試其他的要自己點進去,不給提示的話生成的是caffe.exe。在dos系統中運行爲:

  4. 生成MATLAB接口的方法:首先,進入buildVS2013\MSVCme,需要把matcaffe.vcxproj項目裏面的matlab目錄修改成你自己的(還是用editplus修改即可)。點擊運行matcaffe。成功後會在matlab/+caffe/private文件夾裏面找到一個叫caffe_.mexw64的文件。這就是一個類似MATLAB接口的文件。

Mnist例子測試

雖然成功低編譯出了caffe.exe,但是不用實例來跑一跑的話還是不能證明安裝成功了吧。所以爲了驗證,我用mnist這個例子來試試。
按照我之前參考的博文,大概是這麼個過程:
http://pan.baidu.com/s/1mgl9ndu 下載已經轉換好的MNIST的leveldb數據文件,解壓至./examples/mnist文件夾中,然後運行根目錄下的run_mnist.bat即可開始訓練,訓練日誌會保存在./log文件夾中,以INFO開頭,txt格式的日誌文件中。
沒錯,如果大家這麼就正確運行沒有錯誤了話,恭喜你,後面你就不用看了。現在我要總結下我運行中出現的錯誤。

  1. F1229 14:03:20.650424 72008 cudnn_conv_layer.cpp:53] Check failed: status == CUDNN_STATUS_SUCCESS (6 vs. 0)

    出現這個問題說明計算機的顯卡配置太低,不能用CUDNN加速。CUDNN加速只試用於計算能力3.0以上的,很不幸,我的才2.1。。。
    打開VS2013,將所有工程中c/c++的預處理器定義中的USE_CUDNN刪掉,如下圖

  2. F0105 16:42:30.419942 21836 im2col.cu:57] Check failed: error == cudaSuccess (8 vs. 0) invalid device function

    出現這個問題,是因爲cuda 的設置要改成你自己的計算機顯卡設置
    我之前說過的吧,我的筆記本的GPU計算能力是2.1
    找到cuda的code generation設置,加上compute_20,sm_21
    這裏寫圖片描述
    這個也是所有項目都改過來。

    我之前說過的吧,我的筆記本的GPU計算能力是2.1。好像只有2.1是一個例外,如果是2.0 或3.0或 5.0或 5.2的話,compute和sm 的值都取成一樣的。反正也只有這麼幾個值能取。
    這裏寫圖片描述

    然後,找到你的Makefile.config文件,這是一個example文件。打開它,找到CUDA_ARCH,進行如下修改【當然按照你自己的情況更改就好了】
    這裏寫圖片描述

到這裏遇到的2個錯誤就都解決了。重新編譯所有項目(差不多0.5h+),然後按照我在mnist例子中最開始說的那樣,運行根目錄下的run_mnist.bat。運行時間10-20mins。結果如下:
這裏寫圖片描述

最後祝大家都能成功運行!^_^

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章