diff --git a/docker/xpu.Dockerfile b/docker/xpu.Dockerfile index 90a2414da3f9..fff8cba56f38 100644 --- a/docker/xpu.Dockerfile +++ b/docker/xpu.Dockerfile @@ -48,7 +48,7 @@ RUN --mount=type=secret,id=github_token \ cd /home/sdp && \ . /home/sdp/miniforge3/bin/activate && \ conda activate py${PYTHON_VERSION} && \ - pip3 install torch==2.10.0+xpu torchao torchvision torchaudio==2.10.0+xpu triton-xpu==3.6.0 --index-url https://download.pytorch.org/whl/xpu + pip3 install torch==2.11.0+xpu torchao torchvision torchaudio==2.11.0+xpu --index-url https://download.pytorch.org/whl/xpu RUN --mount=type=secret,id=github_token \ cd /home/sdp && \ @@ -58,7 +58,7 @@ RUN --mount=type=secret,id=github_token \ git clone --branch ${SG_LANG_BRANCH} --single-branch ${SG_LANG_REPO} && \ cd sglang && cd python && \ cp pyproject_xpu.toml pyproject.toml && \ - pip install . && \ + pip install . --extra-index-url https://download.pytorch.org/whl/xpu && \ pip install xgrammar --no-deps && \ pip install msgspec blake3 py-cpuinfo compressed_tensors gguf partial_json_parser einops tabulate --root-user-action=ignore && \ conda install libsqlite=3.48.0 -y && \ diff --git a/docs/platforms/xpu.md b/docs/platforms/xpu.md index 259f3081c47f..4ffa39fe5015 100644 --- a/docs/platforms/xpu.md +++ b/docs/platforms/xpu.md @@ -30,7 +30,7 @@ conda create -n sgl-xpu python=3.12 -y conda activate sgl-xpu # Set PyTorch XPU as primary pip install channel to avoid installing the larger CUDA-enabled version and prevent potential runtime issues. -pip3 install torch==2.10.0+xpu torchao torchvision torchaudio triton-xpu==3.6.0 --index-url https://download.pytorch.org/whl/xpu +pip3 install torch==2.11.0+xpu torchao torchvision torchaudio==2.11.0+xpu --index-url https://download.pytorch.org/whl/xpu pip3 install xgrammar --no-deps # xgrammar will introduce CUDA-enabled triton which might conflict with XPU # Clone the SGLang code diff --git a/python/pyproject_xpu.toml b/python/pyproject_xpu.toml index 315d9e8509a0..821dc88e5a75 100644 --- a/python/pyproject_xpu.toml +++ b/python/pyproject_xpu.toml @@ -15,10 +15,10 @@ classifiers = [ ] dependencies = [ - "torch==2.10.0+xpu", - "torchcodec==0.10.0 ; sys_platform != 'linux' or (sys_platform == 'linux' and platform_machine != 'aarch64' and platform_machine != 'arm64' and platform_machine != 'armv7l')", # torchcodec does not exist in those systems. torch==2.10.0 on XPU uses 0.10.0 + "torch==2.11.0+xpu", + "torchcodec==0.11.0 ; sys_platform != 'linux' or (sys_platform == 'linux' and platform_machine != 'aarch64' and platform_machine != 'arm64' and platform_machine != 'armv7l')", # torchcodec does not exist in those systems. torch==2.11.0 on XPU uses 0.11.0 "av ; sys_platform == 'linux' and (platform_machine == 'aarch64' or platform_machine == 'arm64' or platform_machine == 'armv7l')", - "torchaudio==2.10.0+xpu", + "torchaudio==2.11.0+xpu", "torchvision", "sgl-kernel @ git+https://github.com/sgl-project/sgl-kernel-xpu.git", "IPython", @@ -58,7 +58,7 @@ dependencies = [ "soundfile==0.13.1", "tiktoken", "timm==1.0.16", - "torchao==0.9.0", + "torchao==0.9.0+xpu", "tqdm", "mistral_common>=1.9.0", "transformers==5.3.0",