caffe编译的问题解决:“cublas_v2.h: No such file or directory”

来源:互联网 发布:java string split n 编辑:程序博客网 时间:2024/06/06 04:20


主要问题还是没有把cuda的头文件、库的路径放置到caffe的Makefile.config中;


具体来说:

CUDA7.5中的include、lib路径是安装目录下/usr/local/cuda-7.5/targets/x86_64-linux/下面的include和lib


将其分别添加到caffe根目录下Makefile.config中的"INCLUDE_DIRS"、“LIBRARY_DIRS”后面就可以了。


make all

make test


==========接着在make runtest的时候出现错误 core dumped

1、随机数的那个例子出错,解决:export MKL_CBWR=AUTO

2、test_common.cpp,出现的CUDA launch之类的错误,可能是因为使用了GTX1080以及CUDA7.5的原因,更新CUDA到8.0就没有这个错误了。

2 0
原创粉丝点击