Update Dockerfile

Dockerfile

@@ -8,13 +8,21 @@ RUN apt-get update && apt-get install -y \
     git wget libgl1-mesa-glx libglib2.0-0 ffmpeg libx264-dev \
     build-essential cmake g++ python3-dev
 
+# Install required system dependencies
 RUN apt-get update && apt-get install -y \
-    software-properties-common && \
-    add-apt-repository -y ppa:graphics-drivers/ppa && \
-    apt-get update && apt-get install -y \
+    software-properties-common wget gnupg2 && \
+    rm -rf /var/lib/apt/lists/*
+
+# Add NVIDIA package repositories
+RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb && \
+    dpkg -i cuda-keyring_1.0-1_all.deb && \
+    apt-get update
+
+# Install CUDA libraries manually
+RUN apt-get install -y \
     cuda-command-line-tools-11-7 \
     libcusolver-dev-11-7 \
-    libcublas-dev-11-7 \
+    libcublas-dev \
     libcudnn8
 
 # Create a user