Skip to content

Adding AMD rocm support and cuda support using uv sync --extra cuda/rocm#23

Open
Rouzbehat78 wants to merge 3 commits intofeature/GRPOfrom
feature/amd-cluster
Open

Adding AMD rocm support and cuda support using uv sync --extra cuda/rocm#23
Rouzbehat78 wants to merge 3 commits intofeature/GRPOfrom
feature/amd-cluster

Conversation

@Rouzbehat78
Copy link
Copy Markdown
Contributor

Small adjustment to the pyproject, tested with a smoke test and all package verisoning works.
for AMD cluster:

uv sync --extra rocm
uv pip uninstall vllm torch && uv pip install vllm --extra-index-url https://wheels.vllm.ai/rocm/

for Nvidia cluster:

uv sync --extra cuda

@Rouzbehat78 Rouzbehat78 requested a review from alay2shah April 17, 2026 21:58
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant