From 728818b15c6b28648b9541d88c3690d7e5a77da9 Mon Sep 17 00:00:00 2001 From: Daisuke Nishimatsu Date: Thu, 18 Aug 2022 13:56:45 +0900 Subject: [PATCH 1/3] revert: install each package of tensorrt Signed-off-by: Daisuke Nishimatsu --- ansible/roles/tensorrt/tasks/main.yaml | 23 +++++++++++++++++++++-- 1 file changed, 21 insertions(+), 2 deletions(-) diff --git a/ansible/roles/tensorrt/tasks/main.yaml b/ansible/roles/tensorrt/tasks/main.yaml index e37c128fdc9..8965c14a478 100644 --- a/ansible/roles/tensorrt/tasks/main.yaml +++ b/ansible/roles/tensorrt/tasks/main.yaml @@ -4,7 +4,16 @@ name: - libcudnn8={{ cudnn_version }} - libcudnn8-dev={{ cudnn_version }} - - tensorrt={{ tensorrt_version }} + - libnvinfer8={{ tensorrt_version }} + - libnvinfer-plugin8={{ tensorrt_version }} + - libnvparsers8={{ tensorrt_version }} + - libnvonnxparsers8={{ tensorrt_version }} + - libnvinfer-bin={{ tensorrt_version }} + - libnvinfer-dev={{ tensorrt_version }} + - libnvinfer-plugin-dev={{ tensorrt_version }} + - libnvparsers-dev={{ tensorrt_version }} + - libnvonnxparsers-dev={{ tensorrt_version }} + - libnvinfer-samples={{ tensorrt_version }} allow_downgrade: true update_cache: true @@ -17,4 +26,14 @@ with_items: - libcudnn8 - libcudnn8-dev - - tensorrt + - libnvinfer8 + - libnvinfer-plugin8 + - libnvparsers8 + - libnvonnxparsers8 + - libnvinfer-bin + - libnvinfer-dev + - libnvinfer-plugin-dev + - libnvparsers-dev + - libnvonnxparsers-dev + - libnvinfer-samples + From 1ad5b8cfd25a0b85acb6494c4ee9e60f560e92eb Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 18 Aug 2022 05:00:12 +0000 Subject: [PATCH 2/3] ci(pre-commit): autofix --- ansible/roles/tensorrt/tasks/main.yaml | 1 - 1 file changed, 1 deletion(-) diff --git a/ansible/roles/tensorrt/tasks/main.yaml b/ansible/roles/tensorrt/tasks/main.yaml index 8965c14a478..50b0b71b5bd 100644 --- a/ansible/roles/tensorrt/tasks/main.yaml +++ b/ansible/roles/tensorrt/tasks/main.yaml @@ -36,4 +36,3 @@ - libnvparsers-dev - libnvonnxparsers-dev - libnvinfer-samples - From fe986c383e26706dbda81abd8842ce9e4ebe1466 Mon Sep 17 00:00:00 2001 From: Daisuke Nishimatsu Date: Thu, 18 Aug 2022 14:38:50 +0900 Subject: [PATCH 3/3] fix: update trt version Signed-off-by: Daisuke Nishimatsu --- amd64.env | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/amd64.env b/amd64.env index 64464a8a21c..dad15c2106b 100644 --- a/amd64.env +++ b/amd64.env @@ -4,4 +4,4 @@ base_image=ubuntu:20.04 cuda_base_image=nvidia/cuda:11.6.2-devel-ubuntu20.04 cuda_version=11.6 cudnn_version=8.4.1.50-1+cuda11.6 -tensorrt_version=8.4.2.4-1+cuda11.6 +tensorrt_version=8.4.2-1+cuda11.6