Skip to content

Commit e4a0073

Browse files
sramakintelpre-commit-ci[bot]Tyler Titsworth
authored
IPEX XPU 2.3.110 support (#378)
Signed-off-by: Srikanth Ramakrishna <srikanth.ramakrishna@intel.com> Signed-off-by: Srikanth Ramakrishna <srikanth.ramakrishna@intel.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Tyler Titsworth <tyler.titsworth@intel.com>
1 parent 70bfa56 commit e4a0073

10 files changed

+71
-74
lines changed

‎python/docker-compose.yaml

+1
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ services:
1717
build:
1818
args:
1919
MINIFORGE_VERSION: ${MINIFORGE_VERSION:-Linux-x86_64}
20+
no_proxy: ""
2021
context: .
2122
labels:
2223
dependency.apt.wget: true

‎pytorch/Dockerfile

+10-17
Original file line numberDiff line numberDiff line change
@@ -167,34 +167,27 @@ RUN apt-get update && \
167167
apt-get clean && \
168168
rm -rf /var/lib/apt/lists/*
169169

170-
RUN no_proxy=$no_proxy wget -q -O- https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB \
171-
| gpg --dearmor | tee /usr/share/keyrings/oneapi-archive-keyring.gpg > /dev/null && \
172-
echo "deb [signed-by=/usr/share/keyrings/oneapi-archive-keyring.gpg] https://apt.repos.intel.com/oneapi all main" \
173-
| tee /etc/apt/sources.list.d/oneAPI.list
170+
RUN rm -rf /etc/apt/sources.list.d/intel-gpu-jammy.list
174171

175-
ARG DPCPP_VER
176-
ARG MKL_VER
177-
ARG CCL_VER
172+
ENV OCL_ICD_VENDORS=/etc/OpenCL/vendors
178173

179-
RUN apt-get update && \
180-
apt-get install -y --no-install-recommends --fix-missing \
181-
intel-oneapi-runtime-dpcpp-cpp=${DPCPP_VER} \
182-
intel-oneapi-runtime-mkl=${MKL_VER} \
183-
intel-oneapi-runtime-ccl=${CCL_VER};
174+
FROM ipex-xpu-base AS ipex-xpu-base-wheels-pip
184175

185-
RUN rm -rf /etc/apt/sources.list.d/intel-gpu-jammy.list /etc/apt/sources.list.d/oneAPI.list
176+
WORKDIR /
177+
COPY xpu-requirements.txt .
186178

187-
ENV LD_LIBRARY_PATH=/opt/intel/oneapi/redist/lib:$LD_LIBRARY_PATH
179+
RUN python -m pip install --no-cache-dir -r xpu-requirements.txt && \
180+
rm -rf xpu-requirements.txt
188181

189-
FROM ipex-xpu-base AS ipex-xpu-base-wheels
182+
FROM ipex-xpu-base AS ipex-xpu-base-wheels-idp
190183

191184
WORKDIR /
192185
COPY xpu-requirements.txt .
193186

194-
RUN python -m pip install --no-cache-dir -r xpu-requirements.txt && \
187+
RUN conda run -n idp python -m pip install --no-cache-dir -r xpu-requirements.txt && \
195188
rm -rf xpu-requirements.txt
196189

197-
FROM ipex-xpu-base AS ipex-xpu-jupyter
190+
FROM ipex-xpu-base-wheels-${PACKAGE_OPTION} AS ipex-xpu-jupyter
198191

199192
WORKDIR /jupyter
200193
COPY jupyter-requirements.txt .

‎pytorch/README.md

+17-3
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,8 @@ The images below include support for both CPU and GPU optimizations:
2424

2525
| Tag(s) | Pytorch | IPEX | Driver | Dockerfile |
2626
| ---------------------- | -------- | -------------- | ------ | --------------- |
27-
| `2.1.40-xpu-pip-base`,`2.1.40-xpu` | [v2.1.0] | [v2.1.40+xpu] | [914] | [v0.4.0-Beta] |
27+
| `2.3.110-xpu-pip-base`,`2.3.110-xpu` | [torch-2.3.1] | [v2.3.110+xpu] | [950] | [v0.4.0] |
28+
| `2.1.40-xpu-pip-base`,`2.1.40-xpu` | [v2.1.0] | [v2.1.40+xpu] | [914] | [v0.4.0-Beta] |
2829
| `2.1.30-xpu` | [v2.1.0] | [v2.1.30+xpu] | [803] | [v0.4.0-Beta] |
2930
| `2.1.20-xpu` | [v2.1.0] | [v2.1.20+xpu] | [803] | [v0.3.4] |
3031
| `2.1.10-xpu` | [v2.1.0] | [v2.1.10+xpu] | [736] | [v0.2.3] |
@@ -37,7 +38,7 @@ docker run -it --rm \
3738
--device /dev/dri \
3839
-v /dev/dri/by-path:/dev/dri/by-path \
3940
--ipc=host \
40-
intel/intel-extension-for-pytorch:2.1.40-xpu
41+
intel/intel-extension-for-pytorch:2.3.110-xpu
4142
```
4243

4344
---
@@ -46,6 +47,7 @@ The images below additionally include [Jupyter Notebook](https://jupyter.org/) s
4647

4748
| Tag(s) | Pytorch | IPEX | Driver | Jupyter Port | Dockerfile |
4849
| --------------------- | -------- | ------------- | ------ | ------------ | --------------- |
50+
| `2.3.110-xpu-pip-jupyter` | [torch-2.3.1] | [v2.3.110+xpu] | [950] | `8888` | [v0.4.0-Beta] |
4951
| `2.1.40-xpu-pip-jupyter` | [v2.1.0] | [v2.1.40+xpu] | [914] | `8888` | [v0.4.0-Beta] |
5052
| `2.1.20-xpu-pip-jupyter` | [v2.1.0] | [v2.1.20+xpu] | [803] | `8888` | [v0.3.4] |
5153
| `2.1.10-xpu-pip-jupyter` | [v2.1.0] | [v2.1.10+xpu] | [736] | `8888` | [v0.2.3] |
@@ -57,7 +59,7 @@ docker run -it --rm \
5759
-p 8888:8888 \
5860
--device /dev/dri \
5961
-v /dev/dri/by-path:/dev/dri/by-path \
60-
intel/intel-extension-for-pytorch:2.1.40-xpu-pip-jupyter
62+
intel/intel-extension-for-pytorch:2.3.110-xpu-pip-jupyter
6163
```
6264

6365
After running the command above, copy the URL (something like `http://127.0.0.1:$PORT/?token=***`) into your browser to access the notebook server.
@@ -270,6 +272,12 @@ The images below are [TorchServe*] with CPU Optimizations:
270272

271273
For more details, follow the procedure in the [TorchServe](https://github.com/pytorch/serve/blob/master/examples/intel_extension_for_pytorch/README.md) instructions.
272274

275+
The images below are [TorchServe*] with XPU Optimizations:
276+
277+
| Tag(s) | Pytorch | IPEX | Dockerfile |
278+
| ------------------- | -------- | ------------ | --------------- |
279+
| `2.3.110-serving-xpu` | [torch-2.3.1] | [v2.3.110+xpu] | [v0.4.0-Beta] |
280+
273281
## CPU only images with Intel® Distribution for Python*
274282

275283
The images below are built only with CPU optimizations (GPU acceleration support was deliberately excluded) and include [Intel® Distribution for Python*]:
@@ -308,6 +316,7 @@ The images below are built only with CPU and GPU optimizations and include [Inte
308316

309317
| Tag(s) | Pytorch | IPEX | Driver | Dockerfile |
310318
| ---------------- | -------- | ------------ | -------- | ------ |
319+
| `2.3.110-xpu-idp-base` | [torch-v2.3.1] | [v2.3.110+xpu] | [950] | [v0.4.0-Beta] |
311320
| `2.1.40-xpu-idp-base` | [v2.1.0] | [v2.1.40+xpu] | [914] | [v0.4.0-Beta] |
312321
| `2.1.30-xpu-idp-base` | [v2.1.0] | [v2.1.30+xpu] | [803] | [v0.4.0-Beta] |
313322
| `2.1.10-xpu-idp-base` | [v2.1.0] | [v2.1.10+xpu] | [736] | [v0.2.3] |
@@ -316,6 +325,7 @@ The images below additionally include [Jupyter Notebook](https://jupyter.org/) s
316325

317326
| Tag(s) | Pytorch | IPEX | Driver | Jupyter Port | Dockerfile |
318327
| --------------------- | -------- | ------------- | ------ | ------------ | --------------- |
328+
| `2.3.110-xpu-idp-jupyter` | [torch-v2.3.1] | [v2.3.110+xpu] | [950] | `8888` | [v0.4.0-Beta] |
319329
| `2.1.40-xpu-idp-jupyter` | [v2.1.0] | [v2.1.40+xpu] | [914] | `8888` | [v0.4.0-Beta] |
320330
| `2.1.20-xpu-idp-jupyter` | [v2.1.0] | [v2.1.20+xpu] | [803] | `8888` | [v0.3.4] |
321331
| `2.1.10-xpu-idp-jupyter` | [v2.1.0] | [v2.1.10+xpu] | [736] | `8888` | [v0.2.3] |
@@ -384,6 +394,7 @@ It is the image user's responsibility to ensure that any use of The images below
384394
[v0.2.3]: https://github.com/intel/ai-containers/blob/v0.2.3/pytorch/Dockerfile
385395
[v0.1.0]: https://github.com/intel/ai-containers/blob/v0.1.0/pytorch/Dockerfile
386396
397+
[v2.3.110+xpu]: https://github.com/intel/intel-extension-for-pytorch/releases/tag/v2.3.110%2Bxpu
387398
[v2.1.40+xpu]: https://github.com/intel/intel-extension-for-pytorch/releases/tag/v2.1.40%2Bxpu
388399
[v2.1.30+xpu]: https://github.com/intel/intel-extension-for-pytorch/releases/tag/v2.1.30%2Bxpu
389400
[v2.1.20+xpu]: https://github.com/intel/intel-extension-for-pytorch/releases/tag/v2.1.20%2Bxpu
@@ -399,6 +410,8 @@ It is the image user's responsibility to ensure that any use of The images below
399410
[v2.0.1]: https://github.com/pytorch/pytorch/releases/tag/v2.0.1
400411
[v2.0.0]: https://github.com/pytorch/pytorch/releases/tag/v2.0.0
401412
413+
[torch-v2.3.1]: https://github.com/pytorch/pytorch/tree/v2.3.1
414+
402415
[v3.0]: https://github.com/intel/neural-compressor/releases/tag/v3.0
403416
[v2.6]: https://github.com/intel/neural-compressor/releases/tag/v2.6
404417
[v2.4.1]: https://github.com/intel/neural-compressor/releases/tag/v2.4.1
@@ -422,6 +435,7 @@ It is the image user's responsibility to ensure that any use of The images below
422435
<!-- HuggingFace transformers releases -->
423436
[v4.44.0]: https://github.com/huggingface/transformers/releases/tag/v4.44.0
424437
438+
[950]: https://dgpu-docs.intel.com/releases/stable_950_13_20240814.html
425439
[914]: https://dgpu-docs.intel.com/releases/stable_914_33_20240730.html
426440
[803]: https://dgpu-docs.intel.com/releases/LTS_803.29_20240131.html
427441
[736]: https://dgpu-docs.intel.com/releases/stable_736_25_20231031.html

‎pytorch/docker-compose.yaml

+22-31
Original file line numberDiff line numberDiff line change
@@ -94,13 +94,10 @@ services:
9494
xpu:
9595
build:
9696
args:
97-
CCL_VER: ${CCL_VER:-2021.13.1-31}
98-
DPCPP_VER: ${DPCPP_VER:-2024.2.1-1079}
99-
ICD_VER: ${ICD_VER:-24.22.29735.27-914~22.04}
100-
LEVEL_ZERO_DEV_VER: ${LEVEL_ZERO_DEV_VER:-1.17.6-914~22.04}
101-
LEVEL_ZERO_GPU_VER: ${LEVEL_ZERO_GPU_VER:-1.3.29735.27-914~22.04}
102-
LEVEL_ZERO_VER: ${LEVEL_ZERO_VER:-1.17.6-914~22.04}
103-
MKL_VER: ${MKL_VER:-2024.2.1-103}
97+
ICD_VER: ${ICD_VER:-24.26.30049.10-950~22.04}
98+
LEVEL_ZERO_DEV_VER: ${LEVEL_ZERO_DEV_VER:-1.17.6-950~22.04}
99+
LEVEL_ZERO_GPU_VER: ${LEVEL_ZERO_GPU_VER:-1.3.30049.10-950~22.04}
100+
LEVEL_ZERO_VER: ${LEVEL_ZERO_VER:-1.17.6-950~22.04}
104101
NO_PROXY: ''
105102
PACKAGE_OPTION: ${PACKAGE_OPTION:-pip}
106103
labels:
@@ -109,51 +106,45 @@ services:
109106
dependency.apt.git: true
110107
dependency.apt.gnupg2: true
111108
dependency.apt.gpg-agent: true
112-
dependency.apt.intel-level-zero-gpu: ${LEVEL_ZERO_GPU_VER:-1.3.29735.27-914~22.04}
113-
dependency.apt.intel-oneapi-runtime-ccl: ${CCL_VER:-2021.13.1-31}
114-
dependency.apt.intel-oneapi-runtime-dpcpp-cpp: ${DPCPP_VER:-2024.2.1-1079}
115-
dependency.apt.intel-oneapi-runtime-mkl: ${MKL_VER:-2024.2.1-103}
116-
dependency.apt.intel-opencl-icd: ${ICD_VER:-23.43.27642.40-803~22.04}
117-
dependency.apt.level-zero: ${LEVEL_ZERO_VER:-1.17.6-914~22.04}
118-
dependency.apt.level-zero-dev: ${LEVEL_ZERO_DEV_VER:-1.17.6-914~22.04}
109+
dependency.apt.intel-level-zero-gpu: ${LEVEL_ZERO_GPU_VER:-1.3.30049.10-950~22.04}
110+
dependency.apt.intel-opencl-icd: ${ICD_VER:-24.26.30049.10-950~22.04}
111+
dependency.apt.level-zero: ${LEVEL_ZERO_VER:-1.17.6-950~22.04}
112+
dependency.apt.level-zero-dev: ${LEVEL_ZERO_DEV_VER:-1.17.6-950~22.04}
119113
dependency.apt.rsync: true
120114
dependency.apt.unzip: true
121115
dependency.idp.pip: false
122116
org.opencontainers.base.name: "intel/python:3.10-core"
123117
org.opencontainers.image.title: "Intel® Extension for PyTorch XPU Base Image"
124-
org.opencontainers.image.version: ${IPEX_VERSION:-2.1.40}-xpu-${PACKAGE_OPTION:-pip}-base
125-
target: ipex-xpu-base-wheels
118+
org.opencontainers.image.version: ${IPEX_VERSION:-2.3.110}-xpu-${PACKAGE_OPTION:-pip}-base
119+
target: ipex-xpu-base-wheels-${PACKAGE_OPTION:-pip}
126120
command: >
127121
python -c "import torch;print(torch.device('xpu'));import
128122
intel_extension_for_pytorch as
129-
ipex;print(ipex.xpu.is_available());print(torch.__version__);
123+
ipex;print(torch.xpu.has_xpu());print(torch.__version__);
130124
print(ipex.__version__); [print(f'[{i}]:
131-
{ipex.xpu.get_device_properties(i)}') for i in
132-
range(ipex.xpu.device_count())];"
125+
{torch.xpu.get_device_properties(i)}') for i in
126+
range(torch.xpu.device_count())];"
133127
extends: ipex-base
134-
image: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-ipex-${IPEX_VERSION:-2.1.40xpu}-xpu-base
128+
image: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-ipex-${IPEX_VERSION:-2.3.110xpu}-xpu-base
135129
xpu-jupyter:
136130
build:
137131
args:
138-
CCL_VER: ${CCL_VER:-2021.13.1-31}
139-
DPCPP_VER: ${DPCPP_VER:-2024.2.1-1079}
140-
ICD_VER: ${ICD_VER:-24.22.29735.27-914~22.04}
141-
LEVEL_ZERO_DEV_VER: ${LEVEL_ZERO_DEV_VER:-1.17.6-914~22.04}
142-
LEVEL_ZERO_GPU_VER: ${LEVEL_ZERO_GPU_VER:-1.3.29735.27-914~22.04}
143-
LEVEL_ZERO_VER: ${LEVEL_ZERO_VER:-1.17.6-914~22.04}
144-
MKL_VER: ${MKL_VER:-2024.2.1-103}
132+
ICD_VER: ${ICD_VER:-24.26.30049.10-950~22.04}
133+
LEVEL_ZERO_DEV_VER: ${LEVEL_ZERO_DEV_VER:-1.17.6-950~22.04}
134+
LEVEL_ZERO_GPU_VER: ${LEVEL_ZERO_GPU_VER:-1.3.30049.10-950~22.04}
135+
LEVEL_ZERO_VER: ${LEVEL_ZERO_VER:-1.17.6-950~22.04}
145136
NO_PROXY: ''
146137
PACKAGE_OPTION: ${PACKAGE_OPTION:-pip}
147138
labels:
148139
dependency.python.pip: jupyter-requirements.txt
149-
org.opencontainers.base.name: "intel/intel-optimized-pytorch:${IPEX_VERSION:-2.1.40}-xpu-${PACKAGE_OPTION:-pip}-base"
140+
org.opencontainers.base.name: "intel/intel-optimized-pytorch:${IPEX_VERSION:-2.3.110}-xpu-${PACKAGE_OPTION:-pip}-base"
150141
org.opencontainers.image.title: "Intel® Extension for PyTorch XPU Jupyter Image"
151-
org.opencontainers.image.version: ${IPEX_VERSION:-2.1.40}-xpu-${PACKAGE_OPTION:-pip}-jupyter
142+
org.opencontainers.image.version: ${IPEX_VERSION:-2.3.110}-xpu-${PACKAGE_OPTION:-pip}-jupyter
152143
target: ipex-xpu-jupyter
153144
command: >
154145
bash -c "python -m jupyter --version"
155146
extends: ipex-base
156-
image: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-ipex-${IPEX_VERSION:-2.1.40xpu}-xpu-jupyter
147+
image: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-ipex-${IPEX_VERSION:-2.3.110xpu}-xpu-jupyter
157148
ports:
158149
- 8888:8888
159150
torchserve-cpu:
@@ -196,7 +187,7 @@ services:
196187
docs: serving
197188
org.opencontainers.base.name: "intel/python:3.10-core"
198189
org.opencontainers.image.title: "Intel® Extension for PyTorch XPU Serving Image"
199-
org.opencontainers.image.version: ${IPEX_VERSION:-2.1.40}-serving-xpu
190+
org.opencontainers.image.version: ${IPEX_VERSION:-2.3.110}-serving-xpu
200191
target: torchserve-xpu
201192
command: torchserve --version
202193
entrypoint: ""

‎pytorch/notebooks/ipex-xpu.ipynb

+4-4
Original file line numberDiff line numberDiff line change
@@ -25,13 +25,13 @@
2525
"outputs": [],
2626
"source": [
2727
"import intel_extension_for_pytorch as ipex\n",
28-
"print(ipex.xpu.is_available())\n",
29-
"if (not ipex.xpu.is_available()):\n",
28+
"print(torch.xpu.has_xpu())\n",
29+
"if (not torch.xpu.is_available()):\n",
3030
" print('Intel GPU not detected. Please install GPU with compatible drivers')\n",
3131
" sys.exit(1)\n",
32-
"print(ipex.xpu.has_onemkl())\n",
32+
"print(torch.xpu.has_onemkl())\n",
3333
"print(torch.__version__); print(ipex.__version__)\n",
34-
"[print(f'[{i}]: {ipex.xpu.get_device_properties(i)}') for i in range(ipex.xpu.device_count())]\n"
34+
"[print(f'[{i}]: {torch.xpu.get_device_properties(i)}') for i in range(torch.xpu.device_count())]\n"
3535
]
3636
}
3737
],

‎pytorch/serving/README.md

+2-2
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ docker run --rm -it \
3939
-u root \
4040
-v $PWD:/home/model-server \
4141
--device /dev/dri \
42-
intel/intel-optimized-pytorch:2.1.40-serving-xpu \
42+
intel/intel-optimized-pytorch:2.3.110-serving-xpu \
4343
sh -c 'python model-archive/ipex_squeezenet.py && \
4444
torch-model-archiver --model-name squeezenet1_1 \
4545
--version 1.1 \
@@ -73,7 +73,7 @@ docker run -d --rm --name server \
7373
-v $PWD/config-xpu.properties:/home/model-server/config.properties \
7474
--net=host \
7575
--device /dev/dri \
76-
intel/intel-optimized-pytorch:2.1.40-serving-xpu
76+
intel/intel-optimized-pytorch:2.3.110-serving-xpu
7777
```
7878

7979
After lauching the container, follow the steps below:

‎pytorch/serving/torchserve-xpu-requirements.txt

+4-5
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,8 @@
1-
torch==2.1.0.post3+cxx11.abi
2-
torchvision==0.16.0.post3+cxx11.abi
3-
torchaudio==2.1.0.post3+cxx11.abi
4-
intel_extension_for_pytorch==2.1.40+xpu
1+
torch==2.3.1+cxx11.abi
2+
torchvision==0.18.1+cxx11.abi
3+
torchaudio==2.3.1+cxx11.abi
4+
intel_extension_for_pytorch==2.3.110+xpu
55
--extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us
6-
setuptools==69.5.1
76
numpy==1.26.4
87
captum>=0.7.0
98
cython>=3.0.10

‎pytorch/serving/wf-store/rest-test.sh

+2-2
Original file line numberDiff line numberDiff line change
@@ -33,13 +33,13 @@ apt-get -y install curl
3333

3434
curl --fail -X GET http://localhost:8080/ping
3535

36-
cd ../model-store || exit
36+
cd /home/model-server/model-store || exit
3737
curl --fail -O https://torchserve.pytorch.org/mar_files/cat_dog_classification.mar
3838
curl --fail -O https://torchserve.pytorch.org/mar_files/dog_breed_classification.mar
3939
curl --fail -X POST "http://127.0.0.1:8081/models?url=cat_dog_classification.mar"
4040
curl --fail -X POST "http://127.0.0.1:8081/models?url=dog_breed_classification.mar"
4141

42-
cd ../wf-store || exit
42+
cd /home/model-server/wf-store || exit
4343
curl --fail -X POST "http://127.0.0.1:8081/workflows?url=dog_breed_wf.war"
4444

4545
curl --fail -O https://raw.githubusercontent.com/pytorch/serve/master/examples/Workflows/dog_breed_classification/model_input/Cat.jpg

‎pytorch/tests/tests.yaml

+4-4
Original file line numberDiff line numberDiff line change
@@ -16,14 +16,14 @@ import-ipex-cpu-${PACKAGE_OPTION:-pip}:
1616
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-ipex-${IPEX_VERSION:-2.4.0}-base
1717
cmd: python -c "import torch;import intel_extension_for_pytorch as ipex;print(f'torch {torch.__version__} ipex {ipex.__version__}')"
1818
import-ipex-xpu-${PACKAGE_OPTION:-pip}:
19-
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-ipex-${IPEX_VERSION:-2.1.40xpu}-xpu-base
19+
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-ipex-${IPEX_VERSION:-2.3.110xpu}-xpu-base
2020
cmd: python -c "import torch; import intel_extension_for_pytorch as ipex;[print(f'[{i}] {torch.xpu.get_device_properties(i)}') for i in range(torch.xpu.device_count())];"
2121
device: ["/dev/dri"]
2222
import-cpu-jupyter-${PACKAGE_OPTION:-pip}:
2323
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-ipex-${IPEX_VERSION:-2.4.0}-jupyter
2424
cmd: python -m jupyter --version
2525
import-xpu-jupyter-${PACKAGE_OPTION:-pip}:
26-
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-ipex-${IPEX_VERSION:-2.1.40xpu}-xpu-jupyter
26+
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-ipex-${IPEX_VERSION:-2.3.110xpu}-xpu-jupyter
2727
cmd: python -m jupyter --version
2828
device: ["/dev/dri"]
2929
import-cpu-oneccl-${PACKAGE_OPTION:-pip}:
@@ -46,14 +46,14 @@ ipex-cpu-${PACKAGE_OPTION:-pip}:
4646
- dst: /tests
4747
src: $PWD/pytorch/tests
4848
ipex-xpu-${PACKAGE_OPTION:-pip}:
49-
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-ipex-${IPEX_VERSION:-2.1.40xpu}-xpu-base
49+
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-ipex-${IPEX_VERSION:-2.3.110xpu}-xpu-base
5050
cmd: python /tests/ipex-resnet50.py --ipex --device xpu
5151
device: ["/dev/dri"]
5252
volumes:
5353
- dst: /tests
5454
src: $PWD/pytorch/tests
5555
ipex-xpu-jupyter-${PACKAGE_OPTION:-pip}:
56-
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-ipex-${IPEX_VERSION:-2.1.40xpu}-xpu-jupyter
56+
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-ipex-${IPEX_VERSION:-2.3.110xpu}-xpu-jupyter
5757
cmd: papermill --log-output /jupyter/xpu.ipynb -k python3
5858
device: ["/dev/dri"]
5959
notebook: True

‎pytorch/xpu-requirements.txt

+5-6
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,9 @@
1-
torch==2.1.0.post3+cxx11.abi
2-
torchvision==0.16.0.post3+cxx11.abi
3-
torchaudio==2.1.0.post3+cxx11.abi
4-
intel_extension_for_pytorch==2.1.40+xpu
5-
oneccl_bind_pt==2.1.400+xpu
1+
torch==2.3.1+cxx11.abi
2+
torchvision==0.18.1+cxx11.abi
3+
torchaudio==2.3.1+cxx11.abi
4+
intel_extension_for_pytorch==2.3.110+xpu
5+
oneccl_bind_pt==2.3.100+xpu
66
--extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us
7-
setuptools==69.5.1
87
numpy>=1.26.4
98
idna>=3.7
109
requests>=2.32.0

0 commit comments

Comments
 (0)