Update Dockerfile

Dockerfile

@@ -6,9 +6,18 @@ RUN dpkg --add-architecture amd64
 RUN apt-get update && \
     apt-get install -y --no-install-recommends gnupg2 curl ca-certificates
 RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/7fa2af80.pub
+
+# Import the NVIDIA GPG key
+#RUN apt-get update && apt-get install -y gnupg2 curl
+RUN curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/7fa2af80.pub | gpg --dearmor -o /usr/share/keyrings/cuda-archive-keyring.gpg
+
+# Add the CUDA repository to sources.list
+RUN echo "deb [arch=amd64 signed-by=/usr/share/keyrings/cuda-archive-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64 /" | tee /etc/apt/sources.list.d/cuda.list
+
+
 #RUN echo "deb https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64 /" | sudo tee /etc/apt/sources.list.d/cuda.list
-USER root
-RUN echo "deb https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64 /" > /etc/apt/sources.list.d/cuda.list
+#USER root
+#RUN echo "deb https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64 /" > /etc/apt/sources.list.d/cuda.list
 
 # Install FFmpeg
 RUN apt-get update && \