Skip to content

Commit cd97516

Browse files
ChangLiu0709Xinyu-KangFrankLeeeeezyksir
authored
Added requirements-rocm.txt for AMD GPU and ROCm (#275)
* added requirements-rocm.txt for AMD GPU and ROCm * fixed the workflow on CI runner (#276) * fixed the workflow on CI runner * polish * polish * polish * polish * polish * polish * polish * polish * Added requirements-rocm.txt for the installation support of AMD GPUs and ROCm --------- Co-authored-by: Xinyu-Kang <[email protected]> Co-authored-by: Shenggui Li <[email protected]> Co-authored-by: zyksir <[email protected]>
1 parent ba6159c commit cd97516

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

requirements-rocm.txt

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
# Use the PyTorch ROCm wheel index (choose the stream that matches your system)
2+
--extra-index-url https://download.pytorch.org/whl/rocm6.3
3+
4+
pre-commit
5+
torch==2.8.0+rocm6.3
6+
torchaudio==2.8.0+rocm6.3
7+
torchvision==0.23.0+rocm6.3
8+
transformers==4.57.1
9+
qwen-vl-utils==0.0.11
10+
datasets
11+
setuptools
12+
tqdm
13+
wandb
14+
psutil
15+
numpy
16+
accelerate
17+
pydantic
18+
sglang[all]==0.5.4
19+
openai-harmony

0 commit comments

Comments
 (0)