a9635c964e
* [Feature] enable unsloth on amd gpu * fix the comment --------- Co-authored-by: Daniel Han <danielhanchen@gmail.com>
17 lines
265 B
Text
17 lines
265 B
Text
# Common dependencies
|
|
-r common.txt
|
|
|
|
--extra-index-url https://download.pytorch.org/whl/rocm6.3
|
|
|
|
torch==2.7.0
|
|
torchvision==0.21.0
|
|
torchaudio==2.7.0
|
|
|
|
triton==3.2
|
|
cmake>=3.26,<4
|
|
packaging
|
|
setuptools>=77.0.3,<80.0.0
|
|
setuptools-scm>=8
|
|
wheel
|
|
jinja2>=3.1.6
|
|
amdsmi==6.4.1
|