diff --git a/ci/docker/runtime_functions.sh b/ci/docker/runtime_functions.sh index 2ddbc84a8841..db81dcdf0197 100755 --- a/ci/docker/runtime_functions.sh +++ b/ci/docker/runtime_functions.sh @@ -489,6 +489,8 @@ build_ubuntu_gpu_tensorrt() { build_ccache_wrappers + export LDFLAGS="-Wl,--as-needed" + # Build ONNX pushd . echo "Installing ONNX." @@ -498,7 +500,7 @@ build_ubuntu_gpu_tensorrt() { cd build cmake \ -DCMAKE_CXX_FLAGS=-I/usr/include/python${PYVER}\ - -DBUILD_SHARED_LIBS=ON ..\ + -DBUILD_SHARED_LIBS=OFF ..\ -G Ninja ninja -j 1 -v onnx/onnx.proto ninja -j 1 -v @@ -518,6 +520,7 @@ build_ubuntu_gpu_tensorrt() { mkdir -p /work/mxnet/lib/ cp 3rdparty/onnx-tensorrt/third_party/onnx/build/*.so /work/mxnet/lib/ + cp 3rdparty/onnx-tensorrt/third_party/onnx/build/*.a /work/mxnet/lib/ cp -L 3rdparty/onnx-tensorrt/build/libnvonnxparser_runtime.so.0 /work/mxnet/lib/ cp -L 3rdparty/onnx-tensorrt/build/libnvonnxparser.so.0 /work/mxnet/lib/