agree to uninstall

axolotl-ai-cloud · Apr 22, 2024 · 95c2e5f · 95c2e5f
1 parent 8401b6f
commit 95c2e5f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/Dockerfile-beta b/docker/Dockerfile-beta
@@ -4,7 +4,7 @@ RUN apt update && apt install -y python3.10-venv git-lfs
 
 RUN python3 -m pip install --upgrade pip &&  \
     pip install packaging && \
-    pip uninstall torch-tensorrt && \
+    pip uninstall -y torch-tensorrt && \
     pip install -U torch==2.2.2 && \
     pip install git+https://github.com/NVIDIA/TransformerEngine.git@stable