caffe运行错误: im2col.cu:61] Check failed: error == cudaSuccess (8 vs. 0) invalid device function

Posted BlueOceans

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了caffe运行错误: im2col.cu:61] Check failed: error == cudaSuccess (8 vs. 0) invalid device function相关的知识,希望对你有一定的参考价值。

错误

 im2col.cu:61] Check failed: error == cudaSuccess (8 vs. 0)  invalid device function

原因:由于Makefile.config里面只对cuda8.0一下的架构编译cuda程序,所以采用cuda8.0后cuda程序没有被编译

解决:修改Makefile.config中的 CUDA_ARCH

添加 -gencode arch=compute_61,code=sm_61

计算能力可以参考下面链接添加合适的架构参数

http://blog.csdn.net/allyli0022/article/details/54628987

 

以上是关于caffe运行错误: im2col.cu:61] Check failed: error == cudaSuccess (8 vs. 0) invalid device function的主要内容,如果未能解决你的问题,请参考以下文章

新手试运行了一个 caffe 的 Python 代码,出现这个错误是怎么回事

caffe运行错误 target_blobs.blobs_size()与 source_layer.blobs_size() 不一致

导入 caffe/scipy 错误:liblapack.so.3:未定义符号:sgetrs_N_single

导入caffe / scipy错误:liblapack.so.3:未定义的符号:sgetrs_N_single

caffe2 no gpu support 错误:libprotobuf.so.14 [进行中]

import caffe报错问题