Lock tensorrt to version 10.4.0

osg-htc · Oct 8, 2024 · 5f273ee · 5f273ee
1 parent 12b7205
commit 5f273ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/htc/deeplabcut:3.0.0rc4/Dockerfile b/htc/deeplabcut:3.0.0rc4/Dockerfile
@@ -25,7 +25,7 @@ RUN python3 -m pip install --upgrade \
         numpy \
         tensorflow==2.10 \
         tensorpack==0.11 \
-        tensorrt \
+        tensorrt==10.4.0 \
         tf_slim \
         torch==2.4.1 \
     && \