Faster RCNN CPU安装记录

参考博客:Faster-RCNN+ZF用自己的数据集训练模型(Python版本)

目前执行了2中的(1)(2),由于在make时遇到了错误(无GPU的原因),搜索了下面的博客进行安装

参考博客:Faster R-CNN算法的Caffe实现

  遇到的问题与相关不同:

1、cython编译出错

python setup.py build_ext --inplace
Traceback (most recent call last):
  File "setup.py", line 56, in <module>
    CUDA = locate_cuda()
  File "setup.py", line 44, in locate_cuda
    raise EnvironmentError('The nvcc binary could not be '
EnvironmentError: The nvcc binary could not be located in your $PATH. Either add it to your path, or set $CUDAHOME
Makefile:2: recipe for target 'all' failed
make: *** [all] Error 1
解决方案:

        1)注释掉setup.py中的CUDA = locate_cuda()

      报错:

python setup.py build_ext --inplace
Traceback (most recent call last):
  File "setup.py", line 125, in <module>
    library_dirs=[CUDA['lib64']],
NameError: name 'CUDA' is not defined
Makefile:2: recipe for target 'all' failed
make: *** [all] Error 1

      2)寻找setup中所有与gpu有关的代码 注释掉

#    Extension('nms.gpu_nms',
#        ['nms/nms_kernel.cu', 'nms/gpu_nms.pyx'],
#        library_dirs=[CUDA['lib64']],
#        libraries=['cudart'],
#        language='c++',
#        runtime_library_dirs=[CUDA['lib64']],
#        # this syntax is specific to this build system
#        # we're only going to use certain compiler args with nvcc and not with
#        # gcc the implemen
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值