popov/neuron-tai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ROCm HW support

Browse Source
This commit is contained in:
Dobromir Popov
2026-07-09 01:07:53 +03:00
parent 08826f6ace
commit 1d3d3018cd
5 changed files with 133 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
QUICKSTART.md
View File

@@ -181,7 +181,7 @@ python -m pip install -e packages/tracker -e packages/node -e packages/p2p -e pa
python -m pip install "transformers>=5.12" accelerate safetensors
python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.3
```
<!-- .venv-rocm/bin/pip install --index-url https://rocm.nightlies.amd.com/v2/gfx1151/ --upgrade -->
Keep this separate from a known-good CPU `.venv` until ROCm is verified on that
machine. ROCm wheels are large and host-runtime-sensitive; a failed ROCm install
should not break the CPU fallback environment.