Sync from v0.13
This commit is contained in:
17
requirements/rocm-build.txt
Normal file
17
requirements/rocm-build.txt
Normal file
@@ -0,0 +1,17 @@
|
||||
# Common dependencies
|
||||
-r common.txt
|
||||
|
||||
--extra-index-url https://download.pytorch.org/whl/rocm6.4
|
||||
torch==2.9.0
|
||||
torchvision==0.24.0
|
||||
torchaudio==2.9.0
|
||||
|
||||
triton==3.5.0
|
||||
cmake>=3.26.1,<4
|
||||
packaging>=24.2
|
||||
setuptools>=77.0.3,<80.0.0
|
||||
setuptools-scm>=8
|
||||
wheel
|
||||
jinja2>=3.1.6
|
||||
amdsmi==6.4.3
|
||||
timm>=1.0.17
|
||||
Reference in New Issue
Block a user