faster rcnn CPU mode

最新推荐文章于 2025-01-10 09:28:21 发布

jasonzzj

最新推荐文章于 2025-01-10 09:28:21 发布

阅读量872

点赞数

CC 4.0 BY-SA版权

分类专栏： Python 深度学习

本文链接：https://blog.youkuaiyun.com/jasonzzj/article/details/73920481

Python 同时被 2 个专栏收录

23 篇文章

订阅专栏

深度学习

14 篇文章

订阅专栏

本文介绍如何通过修改setup.py文件来配置基于GPU的非最大抑制(NMS)模块。重点在于使用CUDA进行编译设置，包括指定编译器、库路径、编译选项等，以实现高效的GPU并行计算。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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']]

)