diff --git a/Dockerfile b/Dockerfile
index a8e773f..8c36fb2 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -3,9 +3,8 @@ FROM nvidia/cuda:12.6.0-runtime-ubuntu24.04
 WORKDIR /app
 
 RUN apt-get update && \
-    apt-get install -y python3 python3-pip git && \
+    apt-get install -y python3 python3-venv python3-pip git && \
     ln -s /usr/bin/python3 /usr/bin/python && \
-    pip install --upgrade pip && \
     apt-get clean && rm -rf /var/lib/apt/lists/*
 
 COPY . .