Update Dockerfile

Dockerfile

@@ -7,23 +7,27 @@ ENV MKL_THREADING_LAYER=GNU
 
 RUN apt-get update && apt-get install -y \
     git wget libgl1-mesa-glx libglib2.0-0 ffmpeg libx264-dev \
-    build-essential cmake \
-    g++ python3-dev
+    build-essential cmake g++ python3-dev \
+    cuda-toolkit-11-7
 
 RUN useradd -m -u 1000 user
 
 USER user
 
 ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH \
+    PATH=/home/user/.local/bin:/usr/local/cuda/bin:$PATH \
     PYTHONPATH=$HOME/app \
-	PYTHONUNBUFFERED=1 \
-	GRADIO_ALLOW_FLAGGING=never \
-	GRADIO_NUM_PORTS=1 \
-	GRADIO_SERVER_NAME=0.0.0.0 \
-	GRADIO_THEME=huggingface \
+    PYTHONUNBUFFERED=1 \
+    GRADIO_ALLOW_FLAGGING=never \
+    GRADIO_NUM_PORTS=1 \
+    GRADIO_SERVER_NAME=0.0.0.0 \
+    GRADIO_THEME=huggingface \
     GRADIO_SHARE=False \
-	SYSTEM=spaces
+    SYSTEM=spaces \
+    CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda \
+    CUDA_HOME=/usr/local/cuda \
+    LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH \
+    CMAKE_PREFIX_PATH=/usr/local/cuda
 
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app