From d56d14e566adffab9248d9c857e32e2bd2934d45 Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Fri, 9 Jan 2026 03:07:13 -0800 Subject: [PATCH] [Docs] Improve docs for install on gb200 (#16760) --- docs/get_started/install.md | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/docs/get_started/install.md b/docs/get_started/install.md index f38f958b6..a2329124a 100644 --- 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**