码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2017-08-17 14:22:19      阅读:941      评论:0      收藏:0      [点我收藏+]

标签:dev   config   art   str   sdn   arch   cti   修改   tar   

错误

 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

标签:dev   config   art   str   sdn   arch   cti   修改   tar   

原文地址:http://www.cnblogs.com/haiyang21/p/7381032.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!