faster rcnn CPU mode

来源:互联网 发布:淘宝店免费货源怎么找 编辑:程序博客网 时间:2024/05/17 22:10

cd $FRCN_ROOT/lib, modify setup.py,annotate these code that are correlative GPU, then make


...
#CUDA = locate_cuda()
...
...
#self.set_executable('compiler_so', CUDA['nvcc'])
...
...
#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 implementation of this trick is in customize_compiler() below


extra_compile_args={'gcc': ["-Wno-unused-function"],


'nvcc': ['-arch=sm_35',


'--ptxas-options=-v',


'-c',


'--compiler-options',


"'-fPIC'"]},


include_dirs = [numpy_include, CUDA['include']]


)

原创粉丝点击