部分Caffe 報錯解決方案

1、error: function “atomicAdd(double *, double)” has already been defined

解決方案：

修改該caffe目錄下的common.cuh，在atomicAdd前添加宏判斷即可：

#if !defined(__CUDA_ARCH__) || __CUDA_ARCH__ >= 600

#else

   __device__ double atomicAdd(double* a, double b)

   {

      return b;

   }

#endif

結果：

#ifndef CAFFE_COMMON_CUH_

#define CAFFE_COMMON_CUH_

#include <cuda.h>

#if !defined(__CUDA_ARCH__) || __CUDA_ARCH__ >= 600

#else

// CUDA: atomicAdd is not defined for doubles

static __inline__ __device__ double atomicAdd(double *address, double val) {

unsigned long long int* address_as_ull = (unsigned long long int*)address;

unsigned long long int old = *address_as_ull, assumed;

if (val==0.0)

return __longlong_as_double(old);

do {

assumed = old;

old = atomicCAS(address_as_ull, assumed, __double_as_longlong(val +__longlong_as_double(assumed)));

} while (assumed != old);

return __longlong_as_double(old);

}

#endif

#endif

2、報錯對‘cv::imread(cv::String const&, int)’未定義的引用，需修改Makefile文件

INCLUDE_DIRS += $(BUILD_INCLUDE_DIR) ./src ./include

ifneq ($(CPU_ONLY), 1)

INCLUDE_DIRS += $(CUDA_INCLUDE_DIR)

LIBRARY_DIRS += $(CUDA_LIB_DIR)

LIBRARIES := cudart cublas curand

endif

LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5 opencv_core opencv_highgui opencv_imgproc opencv_imgcodecs

# handle IO dependencies