blob: ebd53e39d8846c06fc77bf7f48d139887217f74c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
--- a/python/CMakeLists.txt
+++ b/python/CMakeLists.txt
@@ -39,7 +39,7 @@ set(CMAKE_CXX_STANDARD ${CPP_STANDARD})
if (NOT MSVC)
# This allows us to use TRT libs shipped with standalone wheels.
- set(CMAKE_SHARED_LINKER_FLAGS -Wl,-rpath=$ORIGIN)
+ #set(CMAKE_SHARED_LINKER_FLAGS -Wl,-rpath=$ORIGIN)
endif()
set(PY_MODULE_NAME tensorrt)
@@ -78,7 +78,7 @@ if (MSVC)
find_path(PY_LIB_DIR ${PYTHON_LIB_NAME}.lib HINTS ${WIN_EXTERNALS}/${PYTHON} ${EXT_PATH}/${PYTHON} PATH_SUFFIXES lib)
message(STATUS "PY_LIB_DIR: ${PY_LIB_DIR}")
else()
- find_path(PY_INCLUDE Python.h HINTS ${EXT_PATH}/${PYTHON} PATH_SUFFIXES include)
+ find_path(PY_INCLUDE Python.h HINTS /usr/lib PATH_SUFFIXES ${PYTHON})
endif()
message(STATUS "PY_INCLUDE: ${PY_INCLUDE}")
@@ -98,7 +98,7 @@ message(STATUS "PY_CONFIG_INCLUDE: ${PY_
# -------- GLOBAL COMPILE OPTIONS --------
include_directories(${TENSORRT_ROOT}/include ${PROJECT_SOURCE_DIR}/include ${CUDA_INCLUDE_DIRS} ${PROJECT_SOURCE_DIR}/docstrings ${ONNX_INC_DIR} ${PYBIND11_DIR})
-link_directories(${TENSORRT_BUILD})
+link_directories(${TENSORRT_BUILD} $ENV{TRT_NONOSS_ROOT}/lib)
if (MSVC)
include_directories(${MSVC_COMPILER_DIR}/include ${MSVC_COMPILER_DIR}/../ucrt/include ${NV_WDKSDK_INC}/um ${NV_WDKSDK_INC}/shared)
@@ -127,7 +127,7 @@ endif()
# -------- START BUILD PROCESS --------
-if (${CMAKE_BUILD_TYPE} STREQUAL Release)
+if (${CMAKE_BUILD_TYPE} STREQUAL Release OR ${CMAKE_BUILD_TYPE} STREQUAL None OR ${CMAKE_BUILD_TYPE} STREQUAL "")
message(STATUS "Building in Release mode")
if (ENABLE_INETWORK_SERIALIZE)
set(TRT_LIBS nvinfer nvonnxparser nvparsers nvinfer_plugin nvserialize)
--- a/python/build.sh
+++ b/python/build.sh
@@ -33,8 +33,9 @@ cmake .. -DCMAKE_BUILD_TYPE=Release \
-DEXT_PATH=${EXT_PATH} \
-DCUDA_INCLUDE_DIRS=${CUDA_ROOT}/include \
-DTENSORRT_ROOT=${ROOT_PATH} \
- -DTENSORRT_BUILD=${ROOT_PATH}/build/
-make -j12
+ -DTENSORRT_BUILD=${ROOT_PATH}/build/ \
+ -DCMAKE_SKIP_RPATH=YES
+make
# Generate wheel
TRT_MAJOR=$(awk '/^\#define NV_TENSORRT_MAJOR/ {print $3}' ${ROOT_PATH}/include/NvInferVersion.h)
|