From 4431fa674860edf95f587997581c651bfd1c91bd Mon Sep 17 00:00:00 2001
From: Ivan Murabito <ivan.murabito@focoos.ai>
Date: Wed, 13 Aug 2025 09:11:41 +0000
Subject: [PATCH] fix: update tensorrt dockerfile to use
 cu12-cudnn9-py312-uv-tensorrt image

---
 .gitignore |  3 +++
 Dockerfile | 10 ++++++++--
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/.gitignore b/.gitignore
index 539bf848..94060719 100644
--- a/.gitignore
+++ b/.gitignore
@@ -98,3 +98,6 @@ site/
 notebooks/test.ipynb
 gradio/output/
 tutorials/experiments
+experiments/
+notebooks/
+wandb/
diff --git a/Dockerfile b/Dockerfile
index 49ac803d..7cad3f5b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -20,5 +20,11 @@ COPY pyproject.toml ./pyproject.toml
 RUN uv pip install --system -e .[onnx]
 
 
-FROM focoos-gpu AS focoos-tensorrt
-RUN uv pip install --system -e .[tensorrt]
+FROM ghcr.io/focoosai/deeplearning:cu12-cudnn9-py312-uv-tensorrt AS focoos-tensorrt
+LABEL authors="focoos.ai"
+
+WORKDIR /app
+
+COPY focoos ./focoos
+COPY pyproject.toml ./pyproject.toml
+RUN uv pip install --system -e .[onnx,tensorrt]