Update Dockerfile

Dockerfile

@@ -5,6 +5,17 @@ ARG DEBIAN_FRONTEND=noninteractive
 # Set up a new user named "user" with user ID 1099
 RUN useradd -m -u 1099 user
 
+# Update packages
+RUN apt-get update && apt-get install -y \
+  build-essential \
+  python3 \
+  python3-pip \
+  git \
+  ffmpeg \
+  libcudnn8 \
+  libcudnn8-dev \
+  sudo
+
 # Switch to the "user" user
 USER user
 
@@ -23,17 +34,6 @@ ENV PYTHONUNBUFFERED=1 \
     GRADIO_ANALYTICS_ENABLED=False
 WORKDIR /app
 
-# Update packages
-RUN apt-get update && apt-get install -y \
-  build-essential \
-  python3 \
-  python3-pip \
-  git \
-  ffmpeg \
-  libcudnn8 \
-  libcudnn8-dev \
-  sudo
-
 # Set up CUDA libraries
 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