18 lines
289 B
Plaintext
18 lines
289 B
Plaintext
# Common dependencies
|
|
-r common.txt
|
|
|
|
|
|
--extra-index-url https://download.pytorch.org/whl/rocm7.1
|
|
torch==2.11.0
|
|
torchvision==0.26.0
|
|
torchaudio==2.11.0
|
|
triton==3.6.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==7.0.2
|
|
timm>=1.0.17
|