Merge pull request #125 from pipeless-ai/fix_tensorrt_image

fix(container): Use base image with CUDA 11.7 for TensorRT
pipeless-ai · Jan 21, 2024 · 8b7c799 · 8b7c799
2 parents ab1a25a + daa8ce6
commit 8b7c799
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/package/Dockerfile-tensorrt b/package/Dockerfile-tensorrt
@@ -1,4 +1,4 @@
-FROM nvcr.io/nvidia/tensorrt:23.08-py3
+FROM nvcr.io/nvidia/tensorrt:22.12-py3
 
 COPY scripts/install_packages /usr/bin/install_packages