1. 程式人生 > >NVCC src/caffe/util/math_functions.cu

NVCC src/caffe/util/math_functions.cu

屏蔽 版本 new caf 需要 nco int try UNC

解決辦法:由於安裝的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

註釋掉

NVCC src/caffe/util/math_functions.cu