Update Dockerfile

Dockerfile

@@ -2,6 +2,12 @@
 # you will also find guides on how best to write your Dockerfile
 FROM debian:bullseye-slim
 
+RUN dpkg --add-architecture amd64
+    apt-get update
+    apt-get install -y --no-install-recommends gnupg2 curl ca-certificates
+    apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/7fa2af80.pub
+    "deb https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64 /" | sudo tee /etc/apt/sources.list.d/cuda.list
+
 # Install FFmpeg
 RUN apt-get update && \
     apt-get install -y python3-pip ffmpeg libcublas-dev