Skip to content

Commit 39c3173

Browse files
committed
added requirements-rocm.txt for AMD GPU and ROCm
1 parent 8731d2c commit 39c3173

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

requirements-rocm.txt

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
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
20+

0 commit comments

Comments
 (0)