fix(container): Use base image with CUDA 11.7 for TensorRT

Signed-off-by: Miguel A. Cabrera Minagorri <[email protected]>
pipeless-ai · Jan 21, 2024 · daa8ce6 · daa8ce6
1 parent ab1a25a
commit daa8ce6
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