update adding a venv

Dockerfile

@@ -5,16 +5,21 @@ FROM nvidia/cuda:12.6.3-cudnn-devel-ubuntu24.04
 #ENV DEBIAN_FRONTEND=noninteractive
 
 # Install Python and pip
-RUN apt-get update && apt-get install -y python3-pip
+RUN apt-get update && apt-get install -y python3-pip python3-venv
 
 # Where we'll copy the code
 WORKDIR /code
 
 # Copy the current directory contents into the container at /code
 COPY ./requirements.txt /code/requirements.txt
-# Install pip requirements
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
-#RUN pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
+
+# Install pip requirements without venv
+#RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Create a virtual environment and install pip requirements
+RUN python3 -m venv /code/venv
+RUN /code/venv/bin/pip install --no-cache-dir --upgrade pip
+RUN /code/venv/bin/pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 # Creates a non-root user with an explicit UID
 # For more info, please refer to https://aka.ms/vscode-docker-python-configure-containers