--install --extra all installs everything: the engine, the model runtime, the trainer and the UI.
- macOS / Linux
- Windows
- pip
--install reads your GPU’s compute capability and installs the matching CUDA build of PyTorch, RTX 50-series included. Everything goes into core/.venv. An environment already active in your shell is left alone.
Update
Pull and re-run the install. It upgrades the engine and the prebuilt UI package together.INLINE_NO_UPDATE_CHECK=1 to turn the check off.
Hardware support
RTX 50-series
RTX 50-series cards need a CUDA 13 build, and--install picks cu130 for you. CUDA 13 needs driver R580 or newer. On an older driver, --install falls back to cu128 and says so. That build works but is frozen at torch 2.11, so update the driver when you can.
To see what your driver reports and which index would be used, without installing anything:
AMD ROCm on Linux
Install normally first, then replace PyTorch, so nothing overwrites your ROCm build afterwards. Pick the index that matches your ROCm version from pytorch.org.None. Don’t run uv sync or --recreate afterwards: both put the default PyTorch back. Either way, open an issue with how it went.
Troubleshooting
The UI is a blank page on Windows. You’re on a build older than v1.3.0. Update and the app sets the right content type itself. The CUDA build is wrong for my card. Name the index yourself, for example--torch-index cu130. This also replaces an installed torch, which a plain re-run does not.
