赞
踩
#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’]]
#),
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。