[Docs] Improve docs for install on gb200 (#16760)

2026-01-09 03:07:13 -08:00
parent 0c4e155a3c
commit d56d14e566
1 changed files with 7 additions and 1 deletions
--- a/docs/get_started/install.md
+++ b/docs/get_started/install.md
@@ -12,7 +12,13 @@ It is recommended to use uv for faster installation:
 ```bash
 pip install --upgrade pip
 pip install uv
-uv pip install "sglang" --prerelease=allow
+uv pip install "sglang"
+```
+
+For different CUDA versions, you can try to add `--extra-index-url` to find the correct [PyTorch](https://pytorch.org/get-started/locally/) wheel.
+For example, on GB200, you will need to do the following. Otherwise, it will install the CPU version of PyTorch.
+```
+uv pip install "sglang" --extra-index-url https://download.pytorch.org/whl/cu129
 ```

 **Quick fixes to common problems**