1. 程式人生 > >ubuntu16.04 cuda8.0 caffe 編譯時遇到 caffe fatal error: cublas_v2.h 沒有那個檔案或目錄

ubuntu16.04 cuda8.0 caffe 編譯時遇到 caffe fatal error: cublas_v2.h 沒有那個檔案或目錄

caffe 配置參考 http://blog.csdn.net/leo_xu06/article/details/53010900

make caffe的過程中碰過  ./include/caffe/util/device_alternate.hpp:34:23: fatal error: cublas_v2.h: 沒有那個檔案或目錄

 查閱了相關部落格後找到了正確的解決方案:

參考:http://blog.csdn.net/vbskj/article/details/52120475

CUDA8.0 中的include、lib路徑是安裝目錄下/usr/local/cuda-8.0下面的include和lib

將其分別新增到caffe根目錄下Makefile.config中的"INCLUDE_DIRS"、“LIBRARY_DIRS”後面就可以了。


將原

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include 
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib 

改為

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/local/cuda-8.0/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/local/cuda-8.0/lib

即可成功編譯