代码拉取完成,页面将自动刷新
from distutils.core import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext
from Cython.Build import cythonize
import numpy
library_dirs = [
'/usr/local/cuda/lib64',
'/usr/local/TensorRT-7.1.3.4/lib', # for my x86_64 PC
'/usr/local/lib',
]
libraries = [
'nvinfer',
'cudnn',
'cublas',
'cudart_static',
'nvToolsExt',
'cudart',
'rt',
]
include_dirs = [
# in case the following numpy include path does not work, you
# could replace it manually with, say,
# '-I/usr/local/lib/python3.6/dist-packages/numpy/core/include',
'-I' + numpy.__path__[0] + '/core/include',
'-I/usr/local/cuda/include',
'-I/usr/local/TensorRT-7.1.3.4/include', # for my x86_64 PC
'-I/usr/local/include',
]
setup(
cmdclass={'build_ext': build_ext},
ext_modules=cythonize(
Extension(
'pytrt',
sources=['pytrt.pyx'],
language='c++',
library_dirs=library_dirs,
libraries=libraries,
extra_compile_args=['-O3', '-std=c++11'] + include_dirs
),
compiler_directives={'language_level': '3'}
)
)
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。