chore: Update Dockerfile to install CUDA 12.4.1 and cuDNN 9.1.1 libraries

Dockerfile

@@ -28,13 +28,17 @@ RUN rm -f /etc/apt/sources.list.d/*.list && \
     python3-venv \
  && rm -rf /var/lib/apt/lists/*
 
+ENV PATH="/home/chatbot/.local/bin:${PATH}"
+
+ RUN wget https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda_12.4.1_550.54.15_linux.run \
+    && sudo sh cuda_12.4.1_550.54.15_linux.run
 
  RUN wget -q https://developer.download.nvidia.com/compute/cudnn/9.1.1/local_installers/cudnn-local-repo-ubuntu2204-9.1.1_1.0-1_amd64.deb \
     && sudo dpkg -i cudnn-local-repo-ubuntu2204-9.1.1_1.0-1_amd64.deb \
     && sudo cp /var/cudnn-local-repo-ubuntu2204-9.1.1/cudnn-*-keyring.gpg /usr/share/keyrings/ \
     && sudo apt-get update && sudo apt-get -y install cudnn 
 
-    # Create a working directory
+# Create a working directory
 WORKDIR /app
 
 # Create a non-root user and switch to it