Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
rocm-build.txt 287 Bytes
Newer Older
Sage Moore's avatar
[ROCm][V1] Add intial ROCm support to V1 (#12790)  
Sage Moore committed
1 
# Common dependencies
Harry Mellor's avatar
Move requirements into their own directory (#12547)  
Harry Mellor committed
2 
-r common.txt
Sage Moore's avatar
[ROCm][V1] Add intial ROCm support to V1 (#12790)  
Sage Moore committed
3
Huy Do's avatar
Update PyTorch to 2.8.0 (#20358)  
Huy Do committed
4
5
6
7 
--extra-index-url https://download.pytorch.org/whl/rocm6.3
torch==2.8.0
torchvision==0.23.0
torchaudio==2.8.0
Sage Moore's avatar
[ROCm][V1] Add intial ROCm support to V1 (#12790)  
Sage Moore committed
8
Burkhard Ringlein's avatar
[Fix] Bump triton version in rocm-build requirements (#21630)  
Burkhard Ringlein committed
9 
triton==3.3.0
Lu Fang's avatar
[Bugfix] Use cmake 3.26.1 instead of 3.26 to avoid build failure (#19019)  
Lu Fang committed
10 
cmake>=3.26.1,<4
Russell Bryant's avatar
[Build] Require setuptools >= 77.0.3 for PEP 639 (#17389)  
Russell Bryant committed
11
12 
packaging>=24.2
setuptools>=77.0.3,<80.0.0
Sage Moore's avatar
[ROCm][V1] Add intial ROCm support to V1 (#12790)  
Sage Moore committed
13
14 
setuptools-scm>=8
wheel
Russell Bryant's avatar
[Build/CI] Upgrade jinja2 to get 3 moderate CVE fixes (#14839)  
Russell Bryant committed
15 
jinja2>=3.1.6
Sage Moore's avatar
[ROCm][V1] Add intial ROCm support to V1 (#12790)  
Sage Moore committed
16 
amdsmi==6.2.4
Concurrensee's avatar
[ROCm] Add dependencies for ROCm (#24900)  
Concurrensee committed
17 
timm>=1.0.17