chore: Update Dockerfile to install CUDA and cuDNN libraries

Dockerfile

@@ -26,8 +26,12 @@ RUN rm -f /etc/apt/sources.list.d/*.list && \
  && rm -rf /var/lib/apt/lists/*
 
 
- 
-# Create a working directory
+ RUN wget 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 cuda cudnn 
+
+    # Create a working directory
 WORKDIR /app
 
 # Create a non-root user and switch to it