File size: 2,588 Bytes
309b968 | 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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 | @echo off
REM ============================================================
REM DaisyChain - Old Hardware Training Pipeline : setup helper
REM ============================================================
setlocal enabledelayedexpansion
cd /d "%~dp0\.."
echo.
echo ____ _ ____ _ _
echo ^| _ \ __ _^(_^)___ _ _^/ ___^| ^|__ __ _^(_^)_ __
echo ^| ^| ^| / _` ^| / __^| ^| ^| ^| ^| ^| '_ \ / _` ^| ^| '_ \
echo ^| ^|_^| ^(_^| ^| \__ \ ^|_^| ^| ^|___^| ^| ^| ^| ^(_^| ^| ^| ^| ^| ^|
echo ^|____/ \__,_^|_^|___/\__, ^|\____^|_^| ^|_^|\__,_^|_^|_^| ^|_^|
echo ^|___/ Old Hardware Training Pipeline
echo.
echo Choose how to run DaisyChain on this machine:
echo [1] Docker (recommended on Windows - most reliable)
echo [2] Python (native; multi-node gloo is unstable on Windows - use WSL/Linux)
echo [3] Just install Python deps
echo [Q] Quit
echo.
set /p choice=" Your choice: "
if /i "%choice%"=="1" goto docker
if /i "%choice%"=="2" goto python
if /i "%choice%"=="3" goto deps
goto end
:docker
where docker >nul 2>nul
if errorlevel 1 (
echo [!] Docker not found. Install Docker Desktop from https://docker.com and re-run.
goto end
)
echo Building and starting the demo cluster ^(3 nodes + dashboard^)...
docker compose -f docker/docker-compose.yml up --build -d
echo.
echo [OK] Cluster starting. Opening the dashboard at http://localhost:8080
timeout /t 4 >nul
start "" http://localhost:8080
echo To stop: docker compose -f docker/docker-compose.yml down
goto end
:deps
where python >nul 2>nul
if errorlevel 1 ( echo [!] Python not found. Install Python 3.9+ first. & goto end )
echo Installing dependencies...
python -m pip install --upgrade pip
python -m pip install torch numpy psutil
python -m pip install -e .
echo [OK] Installed. You can now run: daisychain-train
goto end
:python
call :deps
echo.
echo ---- Configure this node ----
set /p master=" Coordinator IP (MASTER_ADDR, e.g. Tailscale 100.x): "
set /p world=" Total number of machines (WORLD_SIZE): "
set /p rank=" This machine's RANK (0 = coordinator): "
set /p iface=" Network interface (GLOO_SOCKET_IFNAME, e.g. tailscale0): "
set MASTER_ADDR=%master%
set MASTER_PORT=29560
set WORLD_SIZE=%world%
set RANK=%rank%
set GLOO_SOCKET_IFNAME=%iface%
set USE_LIBUV=0
echo.
echo [!] Note: native multi-node training over gloo is unstable on Windows.
echo If it hangs, use the Docker option or run the nodes on Linux/WSL.
echo Launching node RANK=%rank% ...
python -m daisychain.train
goto end
:end
echo.
pause
endlocal
|