Update Dockerfile

Dockerfile

@@ -1,4 +1,9 @@
-FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
+# Use an official PyTorch image with CUDA support as the base image
+FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-devel
+
+# Add NVIDIA CUDA GPG key
+# Add NVIDIA CUDA GPG key using a different key server
+RUN apt-key adv --keyserver hkp://keyserver.ubuntu.com:80 --recv-key A4B469963BF863CC
 
 ENV DEBIAN_FRONTEND=noninteractive
 
@@ -19,16 +24,20 @@ RUN useradd -m -u 1000 user
 # Switch to the "user" user
 USER user
 
+# Set environment variables
 ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH \
+    CUDA_HOME=/usr/local/cuda \
+    PATH=/home/user/.local/bin:$PATH \
+    LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} \
+    LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:${LIBRARY_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
 
 # Set the environment variable to specify the GPU device
 ENV CUDA_DEVICE_ORDER=PCI_BUS_ID