NVCC src/caffe/util/math_functions.cu

Posted 2021-01-13 huhuai

解决办法:由于安装的cuda版本是7.5，当前下载的caffe版本比较新，需要修改里面的makefile文件，屏蔽下面的代码，cuda<8.0

 

 

In the Makefile.example, try commenting out the *_60 and *_61 lines (for compatibility with CUDA < 8.0).

CUDA_ARCH := -gencode arch=compute_20,code=sm_20         -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_52,code=sm_52         #-gencode arch=compute_60,code=sm_60         #-gencode arch=compute_61,code=sm_61         #-gencode arch=compute_61,code=compute_61

注释掉