cuda_success error

問題描述

• 在參考http://www.cnblogs.com/denny402/p/5083300.html自己嘗試預處理和訓練數據的時候，出現了error == cudaSuccess (11 vs. 0) invalid argument的問題

解決方法

• 參考網址：https://github.com/rbgirshick/rcnn/issues/28

• 當時在寫Makefile.config時，把

  CUDA_ARCH := -gencode arch=compute_20,code=sm_20 \ -gencode arch=compute_20,code=sm_21 \
  • 1
  • 2
  • 3

給注釋了，在上面的網址中將這兩行取消注釋，重新編譯一下即可正常訓練和測試了
* 最后的結果

out of memory的問題

• 可以通過設置test的batchsize和test_iter。將batchsize調節得小一點，只要保證兩個相乘之和等于測試的個數即可

使用過程中突然掉電

• 可以從caffemodel中恢復當前狀態并繼續訓練，參考代碼如下：

  ./build/tools/caffe train --solver=models/bvlc_reference_caffenet/solver.prototxt --snapshot=models/bvlc_reference_caffenet/caffenet_train_iter_10000.solverstate
  • 1
  • 2
• 參考鏈接：http://nfeng.cc/2016/05/19/caffe-solverstate/

python接口的使用問題

• 問題：'module' object has no attribute 'LabelMap'
• 解決辦法：需要將python的目錄添加到~/.bashrc中去。
• 參考鏈接：https://github.com/weiliu89/caffe/issues/4

找不到caffe.pb.h的問題

• 我的問題是caffe/util/cudnn.hpp:8:34: fatal error: caffe/proto/caffe.pb.h: 沒有那個文件或目錄

• 在網上找了很多這個問題，但是相同的問題的原因不一樣，我的caffe在make與cmake的過程中都沒有問題，后來是在https://github.com/muupan/dqn-in-the-caffe/issues/3上找到了解決辦法，缺失的文件是放在caffe安裝根目錄的include文件夾中，具體的解決辦法如下

  protoc src/caffe/proto/caffe.proto --cpp_out=. mkdir include/caffe/proto mv src/caffe/proto/caffe.pb.h include/caffe/proto
  • 1
  • 2
  • 3
  • 4

版權聲明：本文為博主原創文章，未經博主允許不得轉載。 https://blog.csdn.net/u012526003/article/details/68215496

總結

以上是生活随笔為你收集整理的caffe使用过程中遇到的问题和解决办法的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。