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 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 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", ], )