From daa8ce6447548e2426ca008df484783d2cbe4f1b Mon Sep 17 00:00:00 2001 From: "Miguel A. Cabrera Minagorri" Date: Sun, 21 Jan 2024 20:06:55 +0100 Subject: [PATCH] fix(container): Use base image with CUDA 11.7 for TensorRT Signed-off-by: Miguel A. Cabrera Minagorri --- package/Dockerfile-tensorrt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/package/Dockerfile-tensorrt b/package/Dockerfile-tensorrt index e24845d..cce1e9b 100644 --- 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