mirror of
https://github.com/stellarshenson/stellars-jupyterhub-ds.git
synced 2026-03-08 06:00:29 +00:00
30 lines
855 B
Bash
Executable File
30 lines
855 B
Bash
Executable File
#!/bin/sh
|
|
CURRENT_FILE=`readlink -f $0`
|
|
CURRENT_DIR=`dirname $CURRENT_FILE`
|
|
cd $CURRENT_DIR
|
|
|
|
# Check if nvidia-smi is available
|
|
if command -v nvidia-smi &> /dev/null; then
|
|
if nvidia-smi > /dev/null 2>&1; then
|
|
echo "Nvidia GPU found."
|
|
# Run the command for when GPU is available
|
|
docker compose --env-file ../.env \
|
|
-f ../compose.yml -f ../compose-gpu.yml \
|
|
up --no-recreate --no-build -d
|
|
else
|
|
echo "Nvidia GPU not found."
|
|
# Run the command for when GPU is not available
|
|
docker compose --env-file ../.env \
|
|
-f ../compose.yml \
|
|
up --no-recreate --no-build -d
|
|
fi
|
|
else
|
|
echo "nvidia-smi command not found. Nvidia GPU not available."
|
|
# Run the command for when GPU is not available
|
|
docker compose --env-file ../.env \
|
|
-f ../compose.yml \
|
|
up --no-recreate --no-build -d
|
|
fi
|
|
|
|
# EOF
|