Skip to content

Commit 4d685f9

Browse files
jikunshangyma11
authored andcommitted
upgrade to pytorch 2.9 (vllm-project#4)
* update dockerfile & req Signed-off-by: Kunshang Ji <kunshang.ji@intel.com> * update wheel name Signed-off-by: Kunshang Ji <kunshang.ji@intel.com> --------- Signed-off-by: Kunshang Ji <kunshang.ji@intel.com>
1 parent 8d498b9 commit 4d685f9

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

docker/Dockerfile.xpu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
FROM intel/deep-learning-essentials:2025.1.3-0-devel-ubuntu24.04 AS vllm-base
1+
FROM intel/deep-learning-essentials:2025.2.2-0-devel-ubuntu24.04 AS vllm-base
22

33
RUN wget -O- https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB | gpg --dearmor | tee /usr/share/keyrings/oneapi-archive-keyring.gpg > /dev/null && \
44
echo "deb [signed-by=/usr/share/keyrings/oneapi-archive-keyring.gpg] https://apt.repos.intel.com/oneapi all main" | tee /etc/apt/sources.list.d/oneAPI.list && \
@@ -25,7 +25,7 @@ RUN apt clean && apt-get update -y && \
2525
RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.12 1
2626
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.12 1
2727

28-
RUN apt install -y libze1 libze-dev libze-intel-gpu1 intel-opencl-icd libze-intel-gpu-raytracing
28+
RUN apt install -y libze1 libze-dev libze-intel-gpu1 intel-opencl-icd libze-intel-gpu-raytracing intel-ocloc
2929

3030
RUN wget https://github.com/uxlfoundation/oneCCL/releases/download/2021.15.6/intel-oneccl-2021.15.6.2_offline.sh
3131
RUN bash intel-oneccl-2021.15.6.2_offline.sh -a --silent --eula accept && echo "source /opt/intel/oneapi/setvars.sh --force" >> /root/.bashrc

requirements/xpu.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,9 @@ datasets # for benchmark scripts
1212
numba == 0.60.0 # v0.61 doesn't support Python 3.9. Required for N-gram speculative decoding
1313
nixl==0.3.0 # for PD disaggregation
1414

15-
torch==2.8.0+xpu
15+
--extra-index-url=https://download.pytorch.org/whl/xpu
16+
torch==2.9.0+xpu
1617
torchaudio
1718
torchvision
18-
--extra-index-url=https://download.pytorch.org/whl/xpu
1919

20-
intel-extension-for-pytorch @ https://intel-extension-for-pytorch.s3.us-east-1.amazonaws.com/ipex_dev/xpu/intel_extension_for_pytorch-2.8.10.post1%2Bxpu-cp312-cp312-linux_x86_64.whl
20+
intel-extension-for-pytorch @ https://ubit-artifactory-ba.intel.com/artifactory/aipc_releases-ba-local/gpu/new/validation/IPEX/weekly/PVC/2025/ww44_Stock_test/py312/intel_extension_for_pytorch-2.9.10+xpu-cp312-cp312-linux_x86_64.whl

0 commit comments

Comments
 (0)