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 diff --git a/ansible/roles/tensorrt/tasks/main.yaml b/ansible/roles/tensorrt/tasks/main.yaml index e37c128fdc9..50b0b71b5bd 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,13 @@ with_items: - libcudnn8 - libcudnn8-dev - - tensorrt + - libnvinfer8 + - libnvinfer-plugin8 + - libnvparsers8 + - libnvonnxparsers8 + - libnvinfer-bin + - libnvinfer-dev + - libnvinfer-plugin-dev + - libnvparsers-dev + - libnvonnxparsers-dev + - libnvinfer-samples