利用caffe框架编译时,遇到caffe error == cudaSuccess (48 vs. 0)的问题

问题描述: 此错误是指自己笔记本显卡算力跟caffe文件里面Makefile.config中有的显卡算力不匹配导致的;解决办法如下:

1 查看显卡型号

nvidia-smi -L

2 获取对应显卡算力

https://en.wikipedia.org/wiki/CUDA#Supported_GPUs

3 修改Makefile.config配置文件

添加对应自己显卡算力

CUDA_ARCH := -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_61,code=sm_61 \
             -gencode arch=compute_75,code=sm_75

             # -gencode arch=compute_20,code=sm_20 \
             # -gencode arch=compute_20,code=sm_21 \

4 重新编译

进入caffe源路径

sudo make clean
sudo make -j8
sudo make distribute

5 代码重新执行