chore: Update Dockerfile to install pciutils for GPU detection

Dockerfile

@@ -3,8 +3,7 @@ FROM nvidia/cuda:11.3.1-base-ubuntu20.04
 ENV DEBIAN_FRONTEND=noninteractive \
     TZ=Europe/Paris
 
-# Remove any third-party apt sources to avoid issues with expiring keys.
-# Install some basic utilities
+# Install some basic utilities and packages for GPU detection
 RUN rm -f /etc/apt/sources.list.d/*.list && \
     apt-get update && apt-get install -y --no-install-recommends \
     curl \
@@ -20,6 +19,7 @@ RUN rm -f /etc/apt/sources.list.d/*.list && \
     build-essential \
     libsndfile-dev \
     software-properties-common \
+    pciutils \
  && rm -rf /var/lib/apt/lists/*
 
 RUN add-apt-repository ppa:flexiondotorg/nvtop && \
@@ -63,11 +63,24 @@ USER root
 
 # User Debian packages
 ## Security warning : Potential user code executed as root (build time)
-RUN --mount=target=/root/packages.txt,source=packages.txt \
-    apt-get update && \
+COPY packages.txt /root/packages.txt
+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
+
+# 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
+
 # Copy notebooks directory to a temporary location
 COPY --chown=user notebooks /home/user/notebooks
 
@@ -78,9 +91,12 @@ COPY --chown=user notebooks /home/user/notebooks
 USER user
 
 # Python packages
-RUN --mount=target=base_requirements.txt,source=base_requirements.txt \
-    --mount=target=notebooks/requirements.txt,source=notebooks/requirements.txt \
-    pip install --no-cache-dir --upgrade -r base_requirements.txt -r notebooks/requirements.txt
+COPY base_requirements.txt /home/user/base_requirements.txt
+COPY notebooks/requirements.txt /home/user/notebooks/requirements.txt
+COPY notebooks/requirements_cpu.txt /home/user/notebooks/requirements_cpu.txt
+COPY notebooks/requirements_gpu.txt /home/user/notebooks/requirements_gpu.txt
+
+RUN pip install --no-cache-dir --upgrade -r /home/user/base_requirements.txt -r /home/user/requirements_specific.txt
 
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app

check_gpu.sh

@@ -0,0 +1,7 @@
+#!/bin/bash
+
+if lspci | grep -i nvidia; then
+    exit 0
+else
+    exit 1
+fi