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

转自:http://blog.csdn.net/u013078356/article/details/51009470


一般情况下都是因为显卡的计算能力不同而导致的(这里假设你前面的步骤都无误),修改 py-faster-rcnn/lib/setup.py 的第135行,将arch改为与你显卡相匹配的数值,(比如我的GTX 760,计算能力是3.0,就将sm_35改成了sm_30)然后删除utils/bbox.c,nms/cpu_nms.c ,nms/gpu_nms.cpp 重新编译即可

我看到有些人说还有其他的问题,那么可以在最开始的makefile.config文件中就开始修改,不过我没有试过,具体步骤如下

As below, there is my solution (thress steps):    
1 if you're using the GPU instance on AWS, then please change the architecture setting into:    
# CUDA architecture setting: going with all of them.    
# For CUDA < 6.0, comment the *_50 lines for compatibility.    
CUDA_ARCH := -gencode arch=compute_30,code=sm_30 \    
-gencode arch=compute_50,code=sm_50 \    
-gencode arch=compute_50,code=compute_50    
Because the GPU in AWS does not support compute_35    
2 I changed sm_35 into sm_30 in lib/setup.py file    
3 cd lib, remove these files: utils/bbox.c nms/cpu_nms.c nms/gpu_nms.cpp, if they exist.    
And then make && cd ../caffe/ && make clean && make -j8 && make pycaffe -j8    


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值