detect GPU during start_server step

Dockerfile

@@ -68,23 +68,10 @@ RUN apt-get update && \
     xargs -r -a /root/packages.txt apt-get install -y --no-install-recommends \
     && rm -rf /var/lib/apt/lists/*
 
-# Script to check for GPU
-COPY check_gpu.sh /root/check_gpu.sh
-RUN chmod +x /root/check_gpu.sh
-
 # Copy notebook requirement files
 COPY notebooks/requirements_cpu.txt /home/user/notebooks/requirements_cpu.txt
 COPY notebooks/requirements_gpu.txt /home/user/notebooks/requirements_gpu.txt
 
-# Conditional installation based on GPU availability
-RUN if /root/check_gpu.sh; then \
-        echo "GPU detected. Installing GPU-specific Python packages." && \
-        cp /home/user/notebooks/requirements_gpu.txt /home/user/requirements_specific.txt; \
-    else \
-        echo "No GPU detected. Installing CPU-specific Python packages." && \
-        cp /home/user/notebooks/requirements_cpu.txt /home/user/requirements_specific.txt; \
-    fi
-
 #######################################
 # End root user section
 #######################################
@@ -93,7 +80,7 @@ USER user
 
 # Python packages
 COPY base_requirements.txt /home/user/base_requirements.txt
-RUN pip install --no-cache-dir --upgrade -r /home/user/base_requirements.txt -r /home/user/requirements_specific.txt
+RUN pip install --no-cache-dir --upgrade -r /home/user/base_requirements.txt
 
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app

start_server.sh

@@ -27,6 +27,15 @@ for file in /home/user/notebooks/**/*; do
     fi
 done
 
+# GPU detection and package installation
+if command -v nvidia-smi &> /dev/null && nvidia-smi -L; then
+    echo "GPU detected. Installing GPU-specific Python packages."
+    pip install --no-cache-dir -r /home/user/notebooks/requirements_gpu.txt
+else
+    echo "No GPU detected. Installing CPU-specific Python packages."
+    pip install --no-cache-dir -r /home/user/notebooks/requirements_cpu.txt
+fi
+
 JUPYTER_TOKEN="${JUPYTER_TOKEN:=huggingface}"
 
 echo "Starting Jupyter Lab with token $JUPYTER_TOKEN"