Update Dockerfile

Dockerfile

@@ -19,6 +19,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     python3.9 \
     python3-pip \
     python3.9-dev \
+    python3.9-venv \
     && rm -rf /var/lib/apt/lists/*
 
 # Set Python 3.9 as the default python and pip versions
@@ -49,8 +50,9 @@ ENV HOME=/home/user \
 # Set working directory
 WORKDIR $HOME/app
 
-# Install PyTorch first
-RUN pip install --no-cache-dir torch==1.7.1 torchvision --extra-index-url https://download.pytorch.org/whl/cu113 && \
+# Upgrade pip and install PyTorch first
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir torch==1.7.1 torchvision --extra-index-url https://download.pytorch.org/whl/cu113 && \
     python -c "import torch; print('PyTorch version:', torch.__version__)"
 
 # Clone the repository