Update Dockerfile

Dockerfile

@@ -4,17 +4,14 @@ ARG DEBIAN_FRONTEND=noninteractive
 
 ENV PYTHONUNBUFFERED=1
 
-RUN apt-get update && apt-get install --no-install-recommends -y \
+RUN apt-get update && apt-get install -y \
   build-essential \
   python3.9 \
   python3-pip \
   git \
-  ffmpeg
-
-RUN apt-get update && apt-get install --no-install-recommends -y \
+  ffmpeg \
   libcudnn8 \
-  libcudnn8-dev \
-
+  libcudnn8-dev 
  
 WORKDIR /code
 
@@ -39,6 +36,7 @@ ENV HOME=/home/user \
 	GRADIO_THEME=huggingface \
 	SYSTEM=spaces
 
+RUN export LD_LIBRARY_PATH=`python3 -c 'import os; import nvidia.cublas.lib; import nvidia.cudnn.lib; import torch; print(os.path.dirname(nvidia.cublas.lib.__file__) + ":" + os.path.dirname(nvidia.cudnn.lib.__file__) + ":" + os.path.dirname(torch.__file__) +"/lib")'`:$LD_LIBRARY_PATH
 RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
 
 # Set the working directory to the user's home directory