Install required packages for example models

triton-inference-server · Oct 11, 2023 · 2e262e1 · 2e262e1
1 parent 3df0787
commit 2e262e1
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/build.py b/build.py
@@ -1355,6 +1355,14 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
     fi
 RUN pip cache purge
 
+# Install required packages for example models
+RUN python3 -m pip install --upgrade pip && \
+        pip3 install transformers && \
+        pip3 install torch && \
+        pip3 install tritonclient[all] && \
+        pip3 install pandas && \
+        pip3 install tabulate
+
 ENV LD_LIBRARY_PATH=/usr/local/tensorrt/lib/:/opt/tritonserver/backends/tensorrtllm:$LD_LIBRARY_PATH
 """