File size: 846 Bytes
7eb3f10 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | #!/usr/bin/env bash
set -euo pipefail
ROOT_DIR="${ROOT_DIR:-/workspace}"
echo "[setup] expected hardware: NVIDIA RTX PRO 6000 Blackwell Server Edition"
echo "[setup] expected OS family: Ubuntu 24.04 / Linux 6.8"
if [ ! -d "$ROOT_DIR/VLAarchtests" ]; then
echo "[setup] expected staged tree at $ROOT_DIR/VLAarchtests" >&2
fi
echo "[setup] base python:"
python --version || true
if [ -x "$ROOT_DIR/.tools/micromamba/bin/micromamba" ]; then
echo "[setup] micromamba envs:"
"$ROOT_DIR/.tools/micromamba/bin/micromamba" env list || true
fi
echo "[setup] current GPU snapshot:"
nvidia-smi --query-gpu=name,driver_version,memory.total --format=csv,noheader || true
echo "[setup] this repo includes package snapshots in environment/"
echo "[setup] recreate the rlbench env at $ROOT_DIR/envs/rlbench before running the overlap baseline path"
|