安装 caffe 命令“make all”时出现错误消息

Question

我跑了

cp Makefile.config.example Makefile.config
make all

按照网站上的建议完成安装。 我使用带有 CUDA 和 OpenBlas 的 Ubuntu 14.04。

错误信息如下所示

CXX/LD -o .build_release/tools/upgrade_net_proto_text.bin .build_release/lib/libcaffe.so：对 caffe::curandGetErrorString(curandStatus)
的未定义引用 .build_release/lib/libcaffe.so：未定义对 caffe::BaseConvolutionLayer::weight_gpu_gemm(double const*, double const*, double*) 的引用
.build_release/lib/libcaffe.so：对 caffe::BaseConvolutionLayer::forward_gpu_bias(double*, double const*)
的未定义引用 .build_release/lib/libcaffe.so：对 caffe::BaseConvolutionLayer::forward_gpu_bias(float*, float const*)
的未定义引用 .build_release/lib/libcaffe.so：未定义对 caffe::cudnn::dataType::zero
的引用 .build_release/lib/libcaffe.so：未定义对 caffe::cudnn::dataType::one
的引用 .build_release/lib/libcaffe.so：未定义对 caffe::BaseConvolutionLayer::backward_gpu_gemm(float const*, float const*, float*)的引用
.build_release/lib/libcaffe.so：对 caffe::cublasGetErrorString(cublasStatus_t)
的未定义引用 .build_release/lib/libcaffe.so：未定义对 caffe::BaseConvolutionLayer::forward_gpu_gemm(double const*, double const*, double*, bool)的引用
.build_release/lib/libcaffe.so：未定义对 caffe::BaseConvolutionLayer::backward_gpu_gemm(double const*, double const*, double*) 的引用
.build_release/lib/libcaffe.so：对 caffe::BaseConvolutionLayer::backward_gpu_bias(double*, double const*)
的未定义引用 .build_release/lib/libcaffe.so：未定义对 caffe::BaseConvolutionLayer::forward_gpu_gemm(float const*, float const*, float*, bool)的引用
.build_release/lib/libcaffe.so：未定义对 caffe::cudnn::dataType::zero
的引用 .build_release/lib/libcaffe.so：未定义对 caffe::BaseConvolutionLayer::weight_gpu_gemm(float const*, float const*, float*)的引用
.build_release/lib/libcaffe.so：未定义对 caffe::BaseConvolutionLayer::backward_gpu_bias(float*, float const*)
的引用 .build_release/lib/libcaffe.so：未定义对 caffe::cudnn::dataType::one
的引用 collect2：错误：ld 返回 1 个退出状态
make: *** [.build_release/tools/upgrade_net_proto_text.bin] 错误 1

我只修改了 Makefile.config。 修改后的 Makefile.config 如下所示

USE_CUDNN := 1
CUSTOM_CXX := g++
CUDA_DIR := /usr/local/cuda
           -gencode arch=compute_20,code=sm_21 \
           -gencode arch=compute_30,code=sm_30 \
           -gencode arch=compute_35,code=sm_35 \
           -gencode arch=compute_50,code=sm_50 \
           -gencode arch=compute_50,code=compute_50
BLAS := OpenBlas
PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/lib/python2.7/dist-packages/numpy/core/include
PYTHON_LIB := /usr/lib
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
BUILD_DIR := build
DISTRIBUTE_DIR := distribute
TEST_GPUID := 0
Q ?= @

Answer 1

您需要将Makefile.config中的BLAS设置更改为

BLAS := open

而不是'OpenBlas'。