nbrightandClaude Opus 5
bcb085754c
SETUP.md sent a fresh machine straight to setup_env.sh, which calls conda and builds inside ~/sum-parts. Neither exists on a clean box, so the first command failed. This repo only carries our scripts; the benchmark itself is a separate clone. bootstrap.sh installs miniconda into $HOME (no sudo), clones tudelft3d/SUM-Parts-Benchmarks, and reports whether the GPU has room for the paper's voxel_max=64000 (~16.5 GB). Idempotent - an existing install or clone is reported and left alone. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
69 lines
2.4 KiB
Bash
69 lines
2.4 KiB
Bash
#!/usr/bin/env bash
|
|
# SUM Parts - bring a bare machine to the point where setup_env.sh can run
|
|
#
|
|
# setup_env.sh and setup_pointnext.sh both assume two things already exist:
|
|
# ~/miniconda3 (they call conda directly)
|
|
# ~/sum-parts (the upstream benchmark clone they build inside)
|
|
#
|
|
# On a fresh machine neither is there, so this runs first. Idempotent: an
|
|
# existing miniconda or clone is left alone.
|
|
#
|
|
# No sudo anywhere. Miniconda installs to $HOME, and the CUDA toolkit comes
|
|
# from conda rather than apt, so this works on locked-down machines.
|
|
set -euo pipefail
|
|
|
|
CONDA_ROOT="$HOME/miniconda3"
|
|
REPO_DIR="$HOME/sum-parts"
|
|
UPSTREAM="https://github.com/tudelft3d/SUM-Parts-Benchmarks.git"
|
|
|
|
echo "=== [1/3] miniconda ==="
|
|
if [ -x "$CONDA_ROOT/bin/conda" ]; then
|
|
echo " present: $($CONDA_ROOT/bin/conda --version)"
|
|
else
|
|
echo " installing to $CONDA_ROOT"
|
|
TMP=$(mktemp -d)
|
|
curl -fsSL -o "$TMP/miniconda.sh" \
|
|
https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
|
|
bash "$TMP/miniconda.sh" -b -p "$CONDA_ROOT"
|
|
rm -rf "$TMP"
|
|
echo " installed: $($CONDA_ROOT/bin/conda --version)"
|
|
fi
|
|
|
|
echo
|
|
echo "=== [2/3] upstream benchmark repo ==="
|
|
if [ -d "$REPO_DIR/.git" ]; then
|
|
echo " present: $REPO_DIR ($(cd "$REPO_DIR" && git log --oneline -1))"
|
|
else
|
|
echo " cloning $UPSTREAM"
|
|
git clone --depth 1 "$UPSTREAM" "$REPO_DIR"
|
|
echo " cloned: $(cd "$REPO_DIR" && git log --oneline -1)"
|
|
fi
|
|
|
|
BUNDLE="$REPO_DIR/semantic_segmentation/PointNeXt_bundle"
|
|
if [ ! -d "$BUNDLE" ]; then
|
|
echo "error: $BUNDLE missing -- upstream layout changed?" >&2
|
|
exit 1
|
|
fi
|
|
echo " bundle: $BUNDLE"
|
|
|
|
echo
|
|
echo "=== [3/3] GPU visible ==="
|
|
if ! command -v nvidia-smi > /dev/null; then
|
|
echo " WARNING: nvidia-smi not found. On WSL2 this means the driver is not"
|
|
echo " exposing the GPU; fix that before continuing -- nothing below works"
|
|
echo " without it."
|
|
else
|
|
nvidia-smi --query-gpu=name,memory.total,driver_version --format=csv,noheader
|
|
TOTAL=$(nvidia-smi --query-gpu=memory.total --format=csv,noheader,nounits | head -1)
|
|
echo
|
|
if [ "$TOTAL" -ge 20000 ]; then
|
|
echo " ${TOTAL} MiB -- the paper setting voxel_max=64000 (needs ~16.5 GB) fits."
|
|
else
|
|
echo " ${TOTAL} MiB -- voxel_max=64000 needs ~16.5 GB and will NOT fit."
|
|
echo " Use scripts/sweep_voxel_max.sh to find the largest value that does."
|
|
fi
|
|
fi
|
|
|
|
echo
|
|
echo "BOOTSTRAP DONE -- next: bash scripts/setup_env.sh"
|