From 4e234b4cf9dd4f90b209ef8db201320723a4ff30 Mon Sep 17 00:00:00 2001 From: gaopengff Date: Fri, 7 Nov 2025 02:34:23 +0800 Subject: [PATCH] [Intel XPU]Update pytorch xpu to 2.9 (#12363) --- docker/xpu.Dockerfile | 4 ++-- docs/platforms/xpu.md | 2 +- python/pyproject_xpu.toml | 4 ++-- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/docker/xpu.Dockerfile b/docker/xpu.Dockerfile index 8030d58cf..5aa57b3d1 100644 --- a/docker/xpu.Dockerfile +++ b/docker/xpu.Dockerfile @@ -3,7 +3,7 @@ # Usage: docker build --build-arg UBUNTU_VERSION=24.04 --build-arg PYTHON_VERSION=3.10 -t sglang:xpu_kernel -f xpu.Dockerfile --no-cache . # Use Intel deep learning essentials base image with Ubuntu 24.04 -FROM intel/deep-learning-essentials:2025.1.3-0-devel-ubuntu24.04 +FROM intel/deep-learning-essentials:2025.2.2-0-devel-ubuntu24.04 # Avoid interactive prompts during package install ENV DEBIAN_FRONTEND=noninteractive @@ -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.8.0+xpu torchao torchvision torchaudio pytorch-triton-xpu==3.4.0 --index-url https://download.pytorch.org/whl/xpu + pip3 install torch==2.9.0+xpu torchao torchvision torchaudio pytorch-triton-xpu==3.5.0 --index-url https://download.pytorch.org/whl/xpu RUN --mount=type=secret,id=github_token \ cd /home/sdp && \ diff --git a/docs/platforms/xpu.md b/docs/platforms/xpu.md index 4b5ecc9e1..099cc413e 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.8.0+xpu torchao torchvision torchaudio pytorch-triton-xpu==3.4.0 --index-url https://download.pytorch.org/whl/xpu +pip3 install torch==2.9.0+xpu torchao torchvision torchaudio pytorch-triton-xpu==3.5.0 --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 4620371e1..2922e4779 100644 --- a/python/pyproject_xpu.toml +++ b/python/pyproject_xpu.toml @@ -15,8 +15,8 @@ classifiers = [ ] dependencies = [ - "torch==2.8.0", - "torchaudio==2.8.0", + "torch==2.9.0", + "torchaudio==2.9.0", "torchvision", "sgl-kernel @ git+https://github.com/sgl-project/sgl-kernel-xpu.git", "IPython",