SPB Git

spb/wp33_uqo

Public MIT
1.1 KB · 18 lines shellscript
1#!/bin/bash2# Prepare a MacLustr node for WP33 experiments: rsync the repo (code only), create a Python 3.12 venv with pinned MLX, run the3# unit tests that do not need a model. Usage: scripts/cluster_setup.sh <alias> [<alias>...]4set -u5SRC=~/Desktop/UQO/Recherche/Recherche-UQO/ia/titan-moe6for n in "$@"; do7  (8    rsync -az --delete --exclude '.venv' --exclude 'third_party' --exclude 'benchmarks/raw' --exclude 'artifacts/private' --exclude '.git' \9      --exclude '__pycache__' --exclude 'paper' "$SRC/" "$n:~/wp33/titan-moe/" || { echo "$n: rsync FAILED"; exit 1; }10    ssh -o ConnectTimeout=15 "$n" 'export PATH=$HOME/.local/bin:/opt/homebrew/bin:$PATH; cd ~/wp33/titan-moe && \11      ( [ -x .venv/bin/python ] || uv venv --python 3.12 .venv -q ) && \12      uv pip install -q --python .venv/bin/python "mlx>=0.32,<0.33" "mlx-lm>=0.31" numpy pytest matplotlib pandas scipy huggingface_hub psutil pyyaml && \13      .venv/bin/python -c "import mlx.core as mx, mlx_lm, platform; print(\"'"$n"'\", platform.mac_ver()[0], \"mlx\", mx.__version__, \"mlx_lm\", mlx_lm.__version__, mx.default_device())"' \14      || echo "$n: venv FAILED"15  ) &16done17wait18
Files · c9b93b1dbc22f1ddf148d7bc23af7b019eec424a
Menu

Theme