"cudaCheckError() failed : invalid device function" Tensorflow: 运行Faster-RCNN中的demo.py报错

运行Tensorflow版本的Faster-RCNN时,报错“cudaCheckError() failed : invalid device function”,而且没有回溯。

GitHub地址:https://github.com/CharlesShang/TFFRCNN     (Tensorflow 版本的Faster-RCNN不只这一个,但内容应该都一样)

百度无果后果断Google,发现这个错误是由于GPU的计算能力设置错误导致的。设置GPU计算能力:

TFFRCNN/lib/make.sh 

nvcc -std=c++11 -c -o psroi_pooling_op.cu.o psroi_pooling_op_gpu.cu.cc \
-I $TF_INC -D GOOGLE_CUDA=1 -x cu -Xcompiler -fPIC -arch=sm_20

TFFRCNN/lib/setup.py

extra_compile_args={'gcc': ["-Wno-unused-function"],
                            'nvcc': ['-arch=sm_20',
                                     '--ptxas-options=-v',
                                     '-c',
                                     '--compiler-options',

                                     "'-fPIC'"]},

GPU计算能力百度一下随便搜,但使用GTX 750 Ti和GTX 750的需要注意:计算能力是2.0而不是5.0!!网上写的都是5.0,这就非常尴尬了。。。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值