Update Dockerfile

Dockerfile

@@ -1,5 +1,4 @@
-#FROM pytorch/pytorch:2.1.2-cuda11.8-cudnn8-devel
-FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-devel
+FROM pytorch/pytorch:2.1.2-cuda11.8-cudnn8-devel
 ENV DEBIAN_FRONTEND=noninteractive
 
 RUN apt-get update && apt-get install -y \
@@ -20,7 +19,7 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/*
 
 # Create a symlink for python
-RUN ln -s /usr/bin/python3.9 /usr/bin/python
+RUN ln -s /usr/bin/python3 /usr/bin/python
 
 # Create a non-root user
 RUN useradd -m -u 1000 user
@@ -42,7 +41,7 @@ ENV HOME=/home/user \
 	SYSTEM=spaces
 
 # Set CUDA_HOME environment variable
-ENV CUDA_HOME=/usr/local/cuda-12.1
+ENV CUDA_HOME=/usr/local/cuda-11.8
 ENV TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6+PTX;8.9;9.0"
 ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 
@@ -107,6 +106,7 @@ RUN pip install --no-cache-dir git+https://github.com/IDEA-Research/GroundingDIN
     pip install --no-cache-dir segment-anything==1.0 && \
     git clone https://github.com/luca-medeiros/lang-segment-anything && \
     cd lang-segment-anything && \
+    git checkout 9bdf77715fa70fca96452184abdc64a6666e0d46 && \  # Replace <commit-hash> with the actual commit hash
     pip install -e . && \
     cd ..