参考博客:Faster-RCNN+ZF用自己的数据集训练模型(Python版本)
目前执行了2中的(1)(2),由于在make时遇到了错误(无GPU的原因),搜索了下面的博客进行安装
遇到的问题与相关不同:
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