From 7498f3002aa923dda10593987b8dbee7804f30c8 Mon Sep 17 00:00:00 2001 From: zzp_miracle Date: Wed, 23 Nov 2022 11:53:27 +0800 Subject: [PATCH 1/3] update tensorrt version to 8.5 --- docker/scripts/install-tensorrt.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/scripts/install-tensorrt.sh b/docker/scripts/install-tensorrt.sh index 75d6feb9ea0..87b96a88585 100644 --- a/docker/scripts/install-tensorrt.sh +++ b/docker/scripts/install-tensorrt.sh @@ -11,7 +11,7 @@ # limitations under the License. set -ex -tensorrt_pkg=TensorRT-8.2.3.0.Linux.x86_64-gnu.cuda-11.4.cudnn8.2.tar.gz +tensorrt_pkg=TensorRT-8.5.1.7.Linux.x86_64-gnu.cuda-11.8.cudnn8.6.tar.gz curl -sL https://pai-blade.oss-accelerate.aliyuncs.com/build_deps/tensorrt/${tensorrt_pkg} -o ${tensorrt_pkg} tar xvfz $tensorrt_pkg -C /usr/local/ 1>/dev/null 2>&1 From 727a75346a6781fc1bfe97df4d624049842d2221 Mon Sep 17 00:00:00 2001 From: zzpmiracle Date: Wed, 23 Nov 2022 17:12:19 +0800 Subject: [PATCH 2/3] fix --- third_party/bazel/tensorrt/trt.BUILD.tpl | 9 --------- 1 file changed, 9 deletions(-) diff --git a/third_party/bazel/tensorrt/trt.BUILD.tpl b/third_party/bazel/tensorrt/trt.BUILD.tpl index 2eabb18a61d..b08ca723380 100644 --- a/third_party/bazel/tensorrt/trt.BUILD.tpl +++ b/third_party/bazel/tensorrt/trt.BUILD.tpl @@ -191,14 +191,6 @@ cc_import( visibility = ["//visibility:private"], ) -cc_import( - name = "proto_static_lib", - static_library = select({ - ":aarch64_linux": "lib/aarch64-linux-gnu/libprotobuf.a", - "//conditions:default": "lib/libprotobuf.a", - }), - visibility = ["//visibility:private"], -) cc_library( name = "nvonnxparser_headers", @@ -245,7 +237,6 @@ cc_library( "nvonnxparser_headers", "nvonnxparser_static_lib", "nvonnx_proto_static_lib", - "proto_static_lib", ], ) From 2c0623e5fea66e59bdea3beb7f6242189d375c1a Mon Sep 17 00:00:00 2001 From: zzp_miracle Date: Wed, 23 Nov 2022 18:55:42 +0800 Subject: [PATCH 3/3] Update pytorch112_gpu.yml --- .github/workflows/pytorch112_gpu.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/pytorch112_gpu.yml b/.github/workflows/pytorch112_gpu.yml index ad2fb8c5ac8..0de56a10b8e 100644 --- a/.github/workflows/pytorch112_gpu.yml +++ b/.github/workflows/pytorch112_gpu.yml @@ -18,7 +18,7 @@ jobs: develop_base_image: nvidia/cuda:11.3.0-cudnn8-devel-ubuntu20.04 runtime_base_image: nvidia/cuda:11.3.0-cudnn8-devel-ubuntu20.04 extra_build_args: --build-arg PYTHON_VERSION=PYTHON3.8 --build-arg ENABLE_FIND_FASTEST_APT_SOURCE=OFF - extra_envs: -e TORCH_BLADE_BUILD_TENSORRT_STATIC=ON + extra_envs: -e TORCH_BLADE_BUILD_TENSORRT_STATIC=OFF -e TORCH_BLADE_CI_BUILD_TORCH_VERSION=1.12.0+cu113 exec_command: bash ./scripts/ci/test_pytorch_blade.sh deploy_command: TORCH_VERSION=1.12.0+cu113 bash ./scripts/ci/deploy_pytorch_blade.sh