Check failed: error == cudaSuccess (8 vs. 0) invalid device function

caffe正常能用，但是运行过程中，报错Check failed: error == cudaSuccess (8 vs. 0)  invalid device function。

事实上，出现这种问题是编译时，显卡的计算能力没有设置好。按照自己的显卡修改Makefile.config的这个位置即可

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_61,code=sm_61

如我用的是1080的显卡，通过下面https://blog.youkuaiyun.com/qq_20481015/article/details/83628605查看对应计算能力。1080对应的是6.1，所以我将上述修改为

CUDA_ARCH := -gencode arch=compute_61,code=sm_61

重新编译，就好了。