Update Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
 
-FROM nvidia/cuda:12.4.1-runtime-ubuntu22.04
+FROM nvidia/cuda:12.4.1-devel-ubuntu22.04
 
 # Set the working directory in the container
 WORKDIR /DocQA
@@ -21,7 +21,7 @@ RUN useradd -m -u 1000 user
 # Switch to the new user
 USER user
 ENV HOME=/home/user
-ENV PATH=/home/user/.local/bin:$PATH
+ENV PATH=/home/user/.local/bin:/usr/local/cuda/bin:$PATH
 
 # Set the working directory for the application
 WORKDIR $HOME/app
@@ -30,7 +30,7 @@ WORKDIR $HOME/app
 COPY --chown=user requirements.txt $HOME/app/
 RUN pip install --no-cache-dir -r requirements.txt
 RUN pip uninstall --y faiss-cpu & pip install faiss-gpu
-RUN pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
+RUN pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu124
 
 # Copy the rest of the application's code to the container
 COPY --chown=user . $HOME/app