Update Dockerfile

Dockerfile

@@ -46,14 +46,9 @@ 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 NVIDIA repository to the APT 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 /" > /etc/apt/sources.list.d/cuda.list
-RUN curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/7fa2af80.pub > 7fa2af80.pub
-RUN gpg --import 7fa2af80.pub
-RUN apt-key del 7fa2af80 \
-    && apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/3bf863cc.pub \
-    && apt-get update \
-    && apt-get install -y libsm6 libgl1-mesa-glx libxext6 libfontconfig1 libxrender1 libcublas-dev\
-    && rm -rf /var/lib/apt/lists/*    && rm -rf /var/lib/apt/lists/*
+RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
+
+RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
 #RUN gpg --list-sigs [email protected]
 #RUN gpg --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/7fa2af80.pub
 
@@ -69,7 +64,7 @@ RUN apt-key del 7fa2af80 \
 
 # Install FFmpeg
 #RUN apt-get update && \
-RUN    apt-get install -y python3-pip ffmpeg
+RUN    apt-get install -y python3-pip ffmpeg libcublas-dev
 
 
 # Use the official Python 3.9 image as the base image