Update Dockerfile

Dockerfile

@@ -23,8 +23,6 @@ USER user
 ENV HOME=/home/user \
     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 \
@@ -34,6 +32,10 @@ ENV HOME=/home/user \
     GRADIO_SHARE=False \
     SYSTEM=spaces
 
+# Set the environment variable to specify the GPU device
+ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
+ENV CUDA_VISIBLE_DEVICES=0
+
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app
 
@@ -49,9 +51,6 @@ RUN mkdir checkpoints
 # Download checkpoint files using aria2
 RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/Vision-CAIR/MiniGPT4-Video/resolve/main/checkpoints/video_llama_checkpoint_last.pth -d $HOME/app/checkpoints -o video_llama_checkpoint_last.pth
 
-# Set the environment variable to specify the GPU device
-ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
-ENV CUDA_VISIBLE_DEVICES=0
 
 # Run your app.py script
 CMD ["python", "minigpt4_video_demo.py"]