Update Dockerfile

Dockerfile

@@ -52,7 +52,7 @@ 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 \
+    && 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 gpg --list-sigs [email protected]
 #RUN gpg --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/7fa2af80.pub
@@ -68,8 +68,8 @@ RUN apt-key del 7fa2af80 \
 #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 && \
-    apt-get install -y python3-pip ffmpeg
+#RUN apt-get update && \
+RUN    apt-get install -y python3-pip ffmpeg
 
 
 # Use the official Python 3.9 image as the base image