1. 程式人生 > >caffe環境搭建及使用教程

caffe環境搭建及使用教程

1、官網:http://caffe.berkeleyvision.org/ 2、環境搭建:https://blog.csdn.net/yggaoeecs/article/details/79163789 3、教程:https://blog.csdn.net/artprog/article/details/79270392 4、書籍:深度學習 Caffe之經典模型詳解與實戰 具體安裝方法可參考前面教程所述,不過安裝前需要確認以下內容(十分重要) 1、cuda是否安裝成功 測試方法: /usr/local/cuda/samples/1_Utilities/deviceQuery$ ./deviceQuery 在這裡插入圖片描述 出現這樣表示通過 2、測試cudnn /usr/src/cudnn_samples_v7/mnistCUDNN$ sudo make /usr/src/cudnn_samples_v7/mnistCUDNN$ ./mnistCUDNN 在這裡插入圖片描述

出現這樣表示通過,具體可參考https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html 常用問題解決辦法: 一、lmdb問題 examples/mnist/convert_mnist_data.cpp:16:18: fatal error: lmdb.h: No such file or directory compilation terminated. 解決辦法: sudo apt-get install liblmdb-dev 然後重新執行make all。 對於Ubuntu16.04,由於安裝的是32位,在後面動態庫連結時仍然會出問題,因此此處應該直接下載原始碼編譯安裝。 二、找不到動態庫問題 /usr/bin/ld: cannot find -lhdf5_hl /usr/bin/ld: cannot find -lhdf5 解決辦法: Makefile.config中第96行新增 /usr/lib/x86_64-linux-gnu/hdf5/serial

/usr/bin/ld: cannot find -llmdb 解決辦法: 下載原始碼,編譯安裝 下載地址: https://github.com/LMDB/lmdb

三、在執行runtest測試時出現有一個測試不通過的情況 [ FAILED ] 1 test, listed below: [ FAILED ] BatchReindexLayerTest/3.TestGradient, where TypeParam = caffe::GPUDevice 1 FAILED TEST Makefile:533: recipe for target ‘runtest’ failed make: *** [runtest] Error 1

解決辦法: 即將Makefile中的這段

… #Debugging ifeq ($(DEBUG), 1) COMMON_FLAGS += -DDEBUG -g -O0 NVCCFLAGS += -G else COMMON_FLAGS += -DNDEBUG -O2 endif …

改為 … #Debugging ifeq ($(DEBUG), 1) COMMON_FLAGS += -DDEBUG -g -O0 NVCCFLAGS += -G else COMMON_FLAGS += -DNDEBUG -O2 NVCCFLAGS += -G endif 改完之後重新編譯即可。