2025-02-13 22:21:50 -08:00
|
|
|
# Common dependencies
|
2025-03-08 17:44:35 +01:00
|
|
|
-r common.txt
|
2025-02-13 22:21:50 -08:00
|
|
|
|
2025-10-21 14:20:18 -07:00
|
|
|
--extra-index-url https://download.pytorch.org/whl/rocm6.4
|
|
|
|
|
torch==2.9.0
|
|
|
|
|
torchvision==0.24.0
|
|
|
|
|
torchaudio==2.9.0
|
2025-02-13 22:21:50 -08:00
|
|
|
|
2025-10-21 14:20:18 -07:00
|
|
|
triton==3.5.0
|
2025-06-03 15:16:17 +08:00
|
|
|
cmake>=3.26.1,<4
|
2025-05-01 02:25:36 -04:00
|
|
|
packaging>=24.2
|
2025-11-12 19:46:58 -05:00
|
|
|
setuptools>=77.0.3,<80.0.0
|
2025-02-13 22:21:50 -08:00
|
|
|
setuptools-scm>=8
|
|
|
|
|
wheel
|
2025-03-15 01:38:19 -04:00
|
|
|
jinja2>=3.1.6
|
2025-11-13 17:24:28 -08:00
|
|
|
amdsmi==6.4.3
|
2025-09-16 14:49:06 -05:00
|
|
|
timm>=1.0.17
|
