Update Dockerfile

Dockerfile

@@ -1,10 +1,13 @@
 # read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
 # you will also find guides on how best to write your Dockerfile
-FROM debian:bullseye
-
-RUN dpkg --add-architecture amd64
-RUN apt-get update && \
-    apt-get install -y --no-install-recommends gnupg2 curl ca-certificates
+FROM debian:bullseye-slim
+ENV NVIDIA_VISIBLE_DEVICES all
+ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
+ENV NVIDIA_REQUIRE_CUDA "cuda>=8.0"
+ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
+#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
@@ -40,30 +43,30 @@ RUN apt-get update && \
 #RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/7fa2af80.pub
 #RUN apt-key adv --recv-keys --keyserver keys.gnupg.net F60F4B3D7FA2AF80
 # Fetch the NVIDIA repository GPG key
-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
+#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 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 gpg --list-sigs [email protected]
-RUN gpg --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/debian11/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
 
 # Update package lists
 
 
-RUN apt-get update
-RUN gpg --delete-key A4B469963BF863CC
+#RUN apt-get update
+#RUN gpg --delete-key A4B469963BF863CC
 
 #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
 
 # Install FFmpeg
-#RUN apt-get update && \
-RUN   apt-get install -y python3-pip ffmpeg libcublas-dev
-ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
+RUN apt-get update && \
+RUN   apt-get install -y python3-pip ffmpeg
+
 
 # Use the official Python 3.9 image as the base image
 #FROM python:3.9