mirror of
https://github.com/NVIDIA/dgx-spark-playbooks.git
synced 2026-04-23 18:33:54 +00:00
33 lines
713 B
Bash
33 lines
713 B
Bash
#!/bin/bash
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd -P)"
|
|
CURR_DIR="$(pwd -P)"
|
|
if [[ "$CURR_DIR" != "$SCRIPT_DIR" ]]; then
|
|
echo "Error: Please run this script from its own directory: $SCRIPT_DIR"
|
|
exit 1
|
|
fi
|
|
|
|
if [[ "$EUID" -eq 0 ]]; then
|
|
echo "Error: This script must not be run as root."
|
|
exit 1
|
|
fi
|
|
|
|
if [[ $# -lt 1 ]]; then
|
|
echo "Usage: bash $0 --help to see the available options"
|
|
exit 1
|
|
fi
|
|
|
|
if [[ ! -d ".venv" ]]; then
|
|
python3 -m venv .venv
|
|
fi
|
|
|
|
source .venv/bin/activate
|
|
|
|
echo "---- Installing required packages ----"
|
|
pip install -r requirements.txt
|
|
|
|
echo "---- Configuring the cluster (args: $*) ----"
|
|
SPARK_CLUSTER_SETUP_WRAPPER=1 python3 ./spark_cluster_setup.py "$@"
|
|
|
|
deactivate
|