[Universal] - reduce size - removed nvidia, triton - Machine Learning Packages

.github/workflows/smoke-universal.yaml

@@ -26,4 +26,4 @@ jobs:
       uses: ./.github/actions/smoke-test
       with:
         image: universal
-        threshold: 14
+        threshold: 10

src/universal/.devcontainer/local-features/machine-learning-packages/install.sh

@@ -29,12 +29,34 @@ sudo_if() {
 
 export DEBIAN_FRONTEND=noninteractive
 
+install_torch_check_gpu() {
+    if ! command -v lspci &> /dev/null; then
+        echo "lspci not found. Attempting to install..."
+        apt-get update && apt-get install -y pciutils
+    fi
+    # Detect GPU
+    set +e
+    GPU=$(lspci | grep -i NVIDIA)
+    set -e
+    if [ -n "$GPU" ]; then
+        echo "GPU Detected. Installing Torch with GPU support."
+        sudo_if /usr/local/python/current/bin/python -m pip install --user --upgrade --no-cache-dir torch
+    else
+        echo "No GPU Detected. Installing Torch with CPU support."
+        sudo_if /usr/local/python/current/bin/python -m pip install --user --upgrade --no-cache-dir torch -f https://download.pytorch.org/whl/cpu/torch_stable.html
+    fi
+}
+
 install_python_package() {
     PACKAGE=${1:-""}
 
     sudo_if /usr/local/python/current/bin/python -m pip uninstall --yes $PACKAGE
     echo "Installing $PACKAGE..."
-    sudo_if /usr/local/python/current/bin/python -m pip install --user --upgrade --no-cache-dir $PACKAGE
+    if [[ $PACKAGE == "torch" ]]; then
+        install_torch_check_gpu
+    else 
+        sudo_if /usr/local/python/current/bin/python -m pip install --user --upgrade --no-cache-dir $PACKAGE
+    fi
 }
 
 if [[ "$(python --version)" != "" ]] && [[ "$(pip --version)" != "" ]]; then