vllm/requirements/rocm-build.txt
Sage Moore 44fa4d556c
[ROCM] Bind triton version to 3.2 in requirements-built.txt (#16664)
Signed-off-by: Sage Moore <sage@neuralmagic.com>
2025-04-16 14:05:28 +08:00

17 lines
254 B
Plaintext

# Common dependencies
-r common.txt
--extra-index-url https://download.pytorch.org/whl/rocm6.2.4
torch==2.6.0
torchvision==0.21.0
torchaudio==2.6.0
triton==3.2
cmake>=3.26,<4
packaging
setuptools>=61
setuptools-scm>=8
wheel
jinja2>=3.1.6
amdsmi==6.2.4