Agnostic cloud gpu docker image and Jupyter lab (#1097)

This commit is contained in:
Wing Lian
2024-01-15 22:37:54 -05:00
committed by GitHub
parent 8487b97cf3
commit ece0211996
4 changed files with 26 additions and 9 deletions

View File

@@ -17,5 +17,16 @@ else
echo "No PUBLIC_KEY ENV variable provided, not starting openSSH daemon"
fi
# Check if JUPYTER_PASSWORD is set and not empty
if [ -n "$JUPYTER_PASSWORD" ]; then
# Set JUPYTER_TOKEN to the value of JUPYTER_PASSWORD
export JUPYTER_TOKEN="$JUPYTER_PASSWORD"
fi
if [ "$JUPYTER_DISABLE" != "1" ]; then
# Run Jupyter Lab in the background
jupyter lab --allow-root --ip 0.0.0.0 &
fi
# Execute the passed arguments (CMD)
exec "$@"