Skip to content

Commit 4a7f238

Browse files
sramakintelsharvil10Tyler Titsworthdependabot[bot]pre-commit-ci[bot]
authored
update ipex layers to 2.1.40-xpu (#305)
Signed-off-by: Srikanth Ramakrishna <srikanth.ramakrishna@intel.com> Signed-off-by: sharvil10 <sharvil.shah@intel.com> Signed-off-by: Tyler Titsworth <tyler.titsworth@intel.com> Signed-off-by: dependabot[bot] <support@github.com> Signed-off-by: tylertitsworth <tyler.titsworth@intel.com> Signed-off-by: Srikanth Ramakrishna <srikanth.ramakrishna@intel.com> Co-authored-by: Sharvil Shah <sharvil.shah@intel.com> Co-authored-by: Tyler Titsworth <tyler.titsworth@intel.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 4dded50 commit 4a7f238

File tree

5 files changed

+53
-64
lines changed

5 files changed

+53
-64
lines changed

‎pytorch/Dockerfile

+4-14
Original file line numberDiff line numberDiff line change
@@ -36,11 +36,6 @@ ARG PYTHON_VERSION
3636
ARG PYTHON_BASE=${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER}-${BASE_IMAGE_NAME}-${BASE_IMAGE_TAG}-${PACKAGE_OPTION}-py${PYTHON_VERSION}-base
3737
FROM ${PYTHON_BASE} AS ipex-base-pip
3838

39-
ARG IPEX_VERSION
40-
ARG PYTORCH_VERSION
41-
ARG TORCHAUDIO_VERSION
42-
ARG TORCHVISION_VERSION
43-
4439
WORKDIR /
4540
COPY requirements.txt .
4641

@@ -49,11 +44,6 @@ RUN python -m pip install --no-cache-dir -r requirements.txt && \
4944

5045
FROM ${PYTHON_BASE} AS ipex-base-idp
5146

52-
ARG IPEX_VERSION
53-
ARG PYTORCH_VERSION
54-
ARG TORCHAUDIO_VERSION
55-
ARG TORCHVISION_VERSION
56-
5747
WORKDIR /
5848
COPY requirements.txt .
5949

@@ -158,8 +148,8 @@ RUN apt-get update && \
158148
rm -rf /var/lib/apt/lists/*
159149

160150
RUN wget -qO - https://repositories.intel.com/gpu/intel-graphics.key | \
161-
gpg --dearmor --output /usr/share/keyrings/intel-graphics.gpg
162-
RUN echo "deb [arch=amd64 signed-by=/usr/share/keyrings/intel-graphics.gpg] https://repositories.intel.com/gpu/ubuntu jammy/lts/2350 unified" | \
151+
gpg --dearmor --yes --output /usr/share/keyrings/intel-graphics.gpg
152+
RUN echo "deb [arch=amd64 signed-by=/usr/share/keyrings/intel-graphics.gpg] https://repositories.intel.com/gpu/ubuntu jammy unified" | \
163153
tee /etc/apt/sources.list.d/intel-gpu-jammy.list
164154

165155
ARG ICD_VER
@@ -171,8 +161,8 @@ RUN apt-get update && \
171161
apt-get install -y --no-install-recommends --fix-missing \
172162
intel-opencl-icd=${ICD_VER} \
173163
intel-level-zero-gpu=${LEVEL_ZERO_GPU_VER} \
174-
level-zero=${LEVEL_ZERO_VER} \
175-
level-zero-dev=${LEVEL_ZERO_DEV_VER} && \
164+
libze1=${LEVEL_ZERO_VER} \
165+
libze-dev=${LEVEL_ZERO_DEV_VER} && \
176166
apt-get clean && \
177167
rm -rf /var/lib/apt/lists/*
178168

‎pytorch/README.md

+12-6
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ 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` | [v2.1.0] | [v2.1.40+xpu] | [914] | [v0.4.0-Beta] |
2728
| `2.1.30-xpu` | [v2.1.0] | [v2.1.30+xpu] | [803] | [v0.4.0-Beta] |
2829
| `2.1.20-xpu` | [v2.1.0] | [v2.1.20+xpu] | [803] | [v0.3.4] |
2930
| `2.1.10-xpu` | [v2.1.0] | [v2.1.10+xpu] | [736] | [v0.2.3] |
@@ -36,7 +37,7 @@ docker run -it --rm \
3637
--device /dev/dri \
3738
-v /dev/dri/by-path:/dev/dri/by-path \
3839
--ipc=host \
39-
intel/intel-extension-for-pytorch:2.1.30-xpu
40+
intel/intel-extension-for-pytorch:2.1.40-xpu
4041
```
4142

4243
---
@@ -45,8 +46,9 @@ The images below additionally include [Jupyter Notebook](https://jupyter.org/) s
4546

4647
| Tag(s) | Pytorch | IPEX | Driver | Jupyter Port | Dockerfile |
4748
| --------------------- | -------- | ------------- | ------ | ------------ | --------------- |
48-
| `2.1.20-xpu-pip-jupyter` | [v2.1.0] | [v2.1.20+xpu] | [803] | `8888` | [v0.3.4] |
49-
| `2.1.10-xpu-pip-jupyter` | [v2.1.0] | [v2.1.10+xpu] | [736] | `8888` | [v0.2.3] |
49+
| `2.1.40-xpu-pip-jupyter` | [v2.1.0] | [v2.1.40+xpu] | [914] | `8888` | [v0.4.0-Beta] |
50+
| `2.1.20-xpu-pip-jupyter` | [v2.1.0] | [v2.1.20+xpu] | [803] | `8888` | [v0.3.4] |
51+
| `2.1.10-xpu-pip-jupyter` | [v2.1.0] | [v2.1.10+xpu] | [736] | `8888` | [v0.2.3] |
5052

5153
### Run the XPU Jupyter Container
5254

@@ -55,7 +57,7 @@ docker run -it --rm \
5557
-p 8888:8888 \
5658
--device /dev/dri \
5759
-v /dev/dri/by-path:/dev/dri/by-path \
58-
intel/intel-extension-for-pytorch:2.1.20-xpu-pip-jupyter
60+
intel/intel-extension-for-pytorch:2.1.40-xpu-pip-jupyter
5961
```
6062

6163
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.
@@ -308,15 +310,17 @@ The images below are built only with CPU and GPU optimizations and include [Inte
308310

309311
| Tag(s) | Pytorch | IPEX | Driver | Dockerfile |
310312
| ---------------- | -------- | ------------ | -------- | ------ |
313+
| `2.1.40-xpu-idp-base` | [v2.1.0] | [v2.1.40+xpu] | [914] | [v0.4.0-Beta] |
311314
| `2.1.30-xpu-idp-base` | [v2.1.0] | [v2.1.30+xpu] | [803] | [v0.4.0-Beta] |
312315
| `2.1.10-xpu-idp-base` | [v2.1.0] | [v2.1.10+xpu] | [736] | [v0.2.3] |
313316

314317
The images below additionally include [Jupyter Notebook](https://jupyter.org/) server:
315318

316319
| Tag(s) | Pytorch | IPEX | Driver | Jupyter Port | Dockerfile |
317320
| --------------------- | -------- | ------------- | ------ | ------------ | --------------- |
318-
| `2.1.20-xpu-idp-jupyter` | [v2.1.0] | [v2.1.20+xpu] | [803] | `8888` | [v0.3.4] |
319-
| `2.1.10-xpu-idp-jupyter` | [v2.1.0] | [v2.1.10+xpu] | [736] | `8888` | [v0.2.3] |
321+
| `2.1.40-xpu-idp-jupyter` | [v2.1.0] | [v2.1.40+xpu] | [914] | `8888` | [v0.4.0-Beta] |
322+
| `2.1.20-xpu-idp-jupyter` | [v2.1.0] | [v2.1.20+xpu] | [803] | `8888` | [v0.3.4] |
323+
| `2.1.10-xpu-idp-jupyter` | [v2.1.0] | [v2.1.10+xpu] | [736] | `8888` | [v0.2.3] |
320324

321325
## Build from Source
322326

@@ -382,6 +386,7 @@ It is the image user's responsibility to ensure that any use of The images below
382386
[v0.2.3]: https://github.com/intel/ai-containers/blob/v0.2.3/pytorch/Dockerfile
383387
[v0.1.0]: https://github.com/intel/ai-containers/blob/v0.1.0/pytorch/Dockerfile
384388
389+
[v2.1.40+xpu]: https://github.com/intel/intel-extension-for-pytorch/releases/tag/v2.1.40%2Bxpu
385390
[v2.1.30+xpu]: https://github.com/intel/intel-extension-for-pytorch/releases/tag/v2.1.30%2Bxpu
386391
[v2.1.20+xpu]: https://github.com/intel/intel-extension-for-pytorch/releases/tag/v2.1.20%2Bxpu
387392
[v2.1.10+xpu]: https://github.com/intel/intel-extension-for-pytorch/releases/tag/v2.1.10%2Bxpu
@@ -419,6 +424,7 @@ It is the image user's responsibility to ensure that any use of The images below
419424
<!-- HuggingFace transformers releases -->
420425
[v4.44.0]: https://github.com/huggingface/transformers/releases/tag/v4.44.0
421426
427+
[914]: https://dgpu-docs.intel.com/releases/stable_914_33_20240730.html
422428
[803]: https://dgpu-docs.intel.com/releases/LTS_803.29_20240131.html
423429
[736]: https://dgpu-docs.intel.com/releases/stable_736_25_20231031.html
424430
[647]: https://dgpu-docs.intel.com/releases/stable_647_21_20230714.html

‎pytorch/docker-compose.yaml

+25-33
Original file line numberDiff line numberDiff line change
@@ -94,38 +94,34 @@ services:
9494
xpu:
9595
build:
9696
args:
97-
CCL_VER: ${CCL_VER:-2021.12.0-309}
98-
DPCPP_VER: ${DPCPP_VER:-2024.1.0-963}
99-
ICD_VER: ${ICD_VER:-23.43.27642.40-803~22.04}
100-
IPEX_XPU_VERSION: ${IPEX_VERSION:-2.1.20}
101-
LEVEL_ZERO_DEV_VER: ${LEVEL_ZERO_DEV_VER:-1.14.0-744~22.04}
102-
LEVEL_ZERO_GPU_VER: ${LEVEL_ZERO_GPU_VER:-1.3.27642.40-803~22.04}
103-
LEVEL_ZERO_VER: ${LEVEL_ZERO_VER:-1.14.0-744~22.04}
104-
MKL_VER: ${MKL_VER:-2024.1.0-691}
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}
105104
NO_PROXY: ''
106-
ONECCL_VERSION: ${ONECCL_VERSION:-2.1.300}
107105
PACKAGE_OPTION: ${PACKAGE_OPTION:-pip}
108-
PYTORCH_XPU_VERSION: ${PYTORCH_VERSION:-2.1.0}
109-
TORCHVISION_XPU_VERSION: ${TORCHVISION_VERSION:-0.16.0}
110106
labels:
111107
dependency.apt.build-essential: true
112108
dependency.apt.clinfo: true
113109
dependency.apt.git: true
114110
dependency.apt.gnupg2: true
115111
dependency.apt.gpg-agent: true
116-
dependency.apt.intel-level-zero-gpu: ${LEVEL_ZERO_GPU_VER:-1.3.27642.40-803~22.04}
117-
dependency.apt.intel-oneapi-runtime-ccl: ${CCL_VER:-2021.12.0-309}
118-
dependency.apt.intel-oneapi-runtime-dpcpp-cpp: ${DPCPP_VER:-2024.1.0-963}
119-
dependency.apt.intel-oneapi-runtime-mkl: ${MKL_VER:-2024.1.0-691}
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}
120116
dependency.apt.intel-opencl-icd: ${ICD_VER:-23.43.27642.40-803~22.04}
121-
dependency.apt.level-zero: ${LEVEL_ZERO_VER:-1.14.0-744~22.04}
122-
dependency.apt.level-zero-dev: ${LEVEL_ZERO_DEV_VER:-1.14.0-744~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}
123119
dependency.apt.rsync: true
124120
dependency.apt.unzip: true
125121
dependency.idp.pip: false
126122
org.opencontainers.base.name: "intel/python:3.10-core"
127123
org.opencontainers.image.title: "Intel® Extension for PyTorch XPU Base Image"
128-
org.opencontainers.image.version: ${IPEX_VERSION:-2.1.20}-xpu-${PACKAGE_OPTION:-pip}-base
124+
org.opencontainers.image.version: ${IPEX_VERSION:-2.1.40}-xpu-${PACKAGE_OPTION:-pip}-base
129125
target: ipex-xpu-base
130126
command: >
131127
python -c "import torch;print(torch.device('xpu'));import
@@ -135,33 +131,29 @@ services:
135131
{ipex.xpu.get_device_properties(i)}') for i in
136132
range(ipex.xpu.device_count())];"
137133
extends: ipex-base
138-
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.30xpu}-xpu-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
139135
xpu-jupyter:
140136
build:
141137
args:
142-
CCL_VER: ${CCL_VER:-2021.12.0-309}
143-
DPCPP_VER: ${DPCPP_VER:-2024.1.0-963}
144-
ICD_VER: ${ICD_VER:-23.43.27642.40-803~22.04}
145-
IPEX_XPU_VERSION: ${IPEX_VERSION:-2.1.20}
146-
LEVEL_ZERO_DEV_VER: ${LEVEL_ZERO_DEV_VER:-1.14.0-744~22.04}
147-
LEVEL_ZERO_GPU_VER: ${LEVEL_ZERO_GPU_VER:-1.3.27642.40-803~22.04}
148-
LEVEL_ZERO_VER: ${LEVEL_ZERO_VER:-1.14.0-744~22.04}
149-
MKL_VER: ${MKL_VER:-2024.1.0-691}
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}
150145
NO_PROXY: ''
151-
ONECCL_VERSION: ${ONECCL_VERSION:-2.1.200}
152146
PACKAGE_OPTION: ${PACKAGE_OPTION:-pip}
153-
PYTORCH_XPU_VERSION: ${PYTORCH_VERSION:-2.1.0}
154-
TORCHVISION_XPU_VERSION: ${TORCHVISION_VERSION:-0.16.0}
155147
labels:
156148
dependency.python.pip: jupyter-requirements.txt
157-
org.opencontainers.base.name: "intel/intel-optimized-pytorch:${IPEX_VERSION:-2.1.20}-xpu-${PACKAGE_OPTION:-pip}-base"
149+
org.opencontainers.base.name: "intel/intel-optimized-pytorch:${IPEX_VERSION:-2.1.40}-xpu-${PACKAGE_OPTION:-pip}-base"
158150
org.opencontainers.image.title: "Intel® Extension for PyTorch XPU Jupyter Image"
159-
org.opencontainers.image.version: ${IPEX_VERSION:-2.1.20}-xpu-${PACKAGE_OPTION:-pip}-jupyter
151+
org.opencontainers.image.version: ${IPEX_VERSION:-2.1.40}-xpu-${PACKAGE_OPTION:-pip}-jupyter
160152
target: ipex-xpu-jupyter
161153
command: >
162154
bash -c "python -m jupyter --version"
163155
extends: ipex-base
164-
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.30xpu}-xpu-jupyter
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
165157
ports:
166158
- 8888:8888
167159
torchserve:

‎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.30xpu}-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.1.40xpu}-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.30xpu}-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.1.40xpu}-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.30xpu}-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.1.40xpu}-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.30xpu}-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.1.40xpu}-xpu-jupyter
5757
cmd: papermill --log-output /jupyter/xpu.ipynb -k python3
5858
device: ["/dev/dri"]
5959
notebook: True

‎pytorch/xpu-requirements.txt

+8-7
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
1-
torch==2.1.0.post2+cxx11.abi
2-
torchvision==0.16.0.post2+cxx11.abi
3-
torchaudio==2.1.0.post2+cxx11.abi
4-
intel_extension_for_pytorch==2.1.30+xpu
5-
oneccl_bind_pt==2.1.300+xpu
6-
--extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
7-
setuptools==71.1.0
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
6+
--extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us
7+
setuptools==69.5.1
8+
numpy==1.26.4

0 commit comments

Comments
 (0)