vllm/requirements/rocm-build.txt
Gregory Shtrasberg a57a3044aa
[ROCm][Build][Bugfix] Bring the base dockerfile in sync with the ROCm fork (#15820)
Signed-off-by: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>
2025-04-01 08:56:39 -07:00

16 lines
242 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
cmake>=3.26,<4
packaging
setuptools>=61
setuptools-scm>=8
wheel
jinja2>=3.1.6
amdsmi==6.2.4