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]