add tensorrt install

microsoft · Dec 21, 2024 · 2375e28 · 2375e28
1 parent 3e638a7
commit 2375e28
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/.azure_pipelines/dockerfiles/linux-gpu.dockerfile b/.azure_pipelines/dockerfiles/linux-gpu.dockerfile
@@ -12,6 +12,15 @@ RUN apt-get update && \
     python${PYTHON_VERSION}-dev \
     python${PYTHON_VERSION}-venv \
     python3-pip \
+    libnvinfer-lean10 \
+    python3-libnvinfer-lean \
+    libnvinfer-dispatch10 \
+    python3-libnvinfer-dispatch \
+    tensorrt-libs \
+    tensorrt-dev \
+    libnvinfer-lean-dev \
+    libnvinfer-dispatch-dev \
+    python3-libnvinfer \
     unzip \
     docker.io
 RUN ln -s /usr/bin/python${PYTHON_VERSION} /usr/bin/python