fix(tensorrt_role): add holding libnvinfer-headers-plugin-dev to fix …

…version conflict (autowarefoundation#4576) * fix(tensorrt_role): add holding libnvinfer-headers-plugin-dev to fix version conflict Signed-off-by: Yuxuan Liu <[email protected]> * add one more Signed-off-by: Yuxuan Liu <[email protected]> * update the order of the readme file to look more neat Signed-off-by: Yuxuan Liu <[email protected]> --------- Signed-off-by: Yuxuan Liu <[email protected]> Co-authored-by: Yuxuan Liu <[email protected]>
tier4 · Mar 29, 2024 · 56470d5 · 56470d5
1 parent 59a4a95
commit 56470d5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/ansible/roles/tensorrt/README.md b/ansible/roles/tensorrt/README.md
@@ -37,6 +37,8 @@ sudo apt-get install -y \
 libcudnn8-dev=${cudnn_version} \
 libnvinfer-dev=${tensorrt_version} \
 libnvinfer-plugin-dev=${tensorrt_version} \
+libnvinfer-headers-dev=${tensorrt_version} \
+libnvinfer-headers-plugin-dev=${tensorrt_version} \
 libnvparsers-dev=${tensorrt_version} \
 libnvonnxparsers-dev=${tensorrt_version}
 
@@ -45,5 +47,7 @@ libcudnn8-dev \
 libnvinfer-dev \
 libnvinfer-plugin-dev \
 libnvparsers-dev \
-libnvonnxparsers-dev
+libnvonnxparsers-dev \
+libnvinfer-headers-dev \
+libnvinfer-headers-plugin-dev
 ```
diff --git a/ansible/roles/tensorrt/tasks/main.yaml b/ansible/roles/tensorrt/tasks/main.yaml
@@ -18,6 +18,8 @@
       - libcudnn8-dev={{ cudnn_version }}
       - libnvinfer-dev={{ tensorrt_version }}
       - libnvinfer-plugin-dev={{ tensorrt_version }}
+      - libnvinfer-headers-dev={{ tensorrt_version }}
+      - libnvinfer-headers-plugin-dev={{ tensorrt_version }}
       - libnvparsers-dev={{ tensorrt_version }}
       - libnvonnxparsers-dev={{ tensorrt_version }}
     allow_change_held_packages: true
@@ -47,6 +49,8 @@
     - libcudnn8-dev
     - libnvinfer-dev
     - libnvinfer-plugin-dev
+    - libnvinfer-headers-dev
+    - libnvinfer-headers-plugin-dev
     - libnvparsers-dev
     - libnvonnxparsers-dev
   when: install_devel == 'true'