Skip to content

Commit fb5fce5

Browse files
authored
Adopt rc2 wheels (#4756)
1 parent 8d4670e commit fb5fce5

File tree

3 files changed

+32
-35
lines changed

3 files changed

+32
-35
lines changed

docker/Dockerfile.prebuilt

Lines changed: 1 addition & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -58,22 +58,7 @@ RUN apt-get update && \
5858
apt-get clean && \
5959
rm -rf /var/lib/apt/lists/*
6060

61-
RUN wget -O- https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB \
62-
| gpg --dearmor | tee /usr/share/keyrings/oneapi-archive-keyring.gpg > /dev/null && \
63-
echo "deb [signed-by=/usr/share/keyrings/oneapi-archive-keyring.gpg] https://apt.repos.intel.com/oneapi all main" \
64-
| tee /etc/apt/sources.list.d/oneAPI.list
65-
66-
ARG DPCPP_VER
67-
ARG MKL_VER
68-
ARG CCL_VER
69-
70-
RUN apt-get update && \
71-
apt-get install -y --no-install-recommends \
72-
intel-oneapi-runtime-dpcpp-cpp=${DPCPP_VER} \
73-
intel-oneapi-runtime-mkl=${MKL_VER} \
74-
intel-oneapi-runtime-ccl=${CCL_VER} && \
75-
apt-get clean && \
76-
rm -rf /var/lib/apt/lists/*
61+
ENV OCL_ICD_VENDORS=/etc/OpenCL/vendors
7762

7863
ARG PYTHON
7964
RUN apt-get update && apt install -y software-properties-common

docker/README.md

Lines changed: 31 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -33,18 +33,20 @@ Alternatively, `./build.sh` script has docker build command to install prebuilt
3333
To pull docker images use the following command:
3434

3535
```bash
36-
docker pull intel/intel-extension-for-pytorch:xpu
36+
docker pull intel/intel-extension-for-pytorch:2.3.110-xpu
3737
```
38+
3839
### Running container:
3940

4041
Run the following commands to start Intel® Extension for PyTorch\* GPU container. You can use `-v` option to mount your
4142
local directory into the container. The `-v` argument can be omitted if you do not need
4243
access to a local directory in the container. Pass the video and render groups to your
4344
docker container so that the GPU is accessible.
4445

46+
```bash
47+
IMAGE_NAME=intel/intel-extension-for-pytorch:2.3.110-xpu
4548
```
46-
IMAGE_NAME=intel/intel-extension-for-pytorch:xpu
47-
```
49+
4850
```bash
4951
docker run --rm \
5052
-v <your-local-dir>:/workspace \
@@ -58,44 +60,57 @@ docker run --rm \
5860

5961
#### Verify if XPU is accessible from PyTorch:
6062
You are inside the container now. Run the following command to verify XPU is visible to PyTorch:
63+
6164
```bash
6265
python -c "import torch;print(torch.device('xpu'))"
6366
```
67+
6468
Sample output looks like below:
65-
```
69+
70+
```bash
6671
xpu
6772
```
73+
6874
Then, verify that the XPU device is available to Intel® Extension for PyTorch\*:
75+
6976
```bash
70-
python -c "import intel_extension_for_pytorch as ipex;print(ipex.xpu.is_available())"
77+
python -c "import torch;import intel_extension_for_pytorch as ipex;print(torch.xpu.has_xpu())"
7178
```
79+
7280
Sample output looks like below:
73-
```
81+
82+
```bash
7483
True
7584
```
85+
7686
Use the following command to check whether MKL is enabled as default:
87+
7788
```bash
78-
python -c "import intel_extension_for_pytorch as ipex;print(ipex.xpu.has_onemkl())"
89+
python -c "import torch;import intel_extension_for_pytorch as ipex;print(torch.xpu.has_onemkl())"
7990
```
91+
8092
Sample output looks like below:
81-
```
93+
94+
```bash
8295
True
8396
```
97+
8498
Finally, use the following command to show detailed info of detected device:
99+
85100
```bash
86-
python -c "import torch; import intel_extension_for_pytorch as ipex; print(torch.__version__); print(ipex.__version__); [print(f'[{i}]: {ipex.xpu.get_device_properties(i)}') for i in range(ipex.xpu.device_count())];"
101+
python -c "import torch; import intel_extension_for_pytorch as ipex; print(torch.__version__); print(ipex.__version__); [print(f'[{i}]: {torch.xpu.get_device_properties(i)}') for i in range(torch.xpu.device_count())];"
87102
```
88103

89104
Sample output looks like below:
105+
106+
```bash
107+
2.3.1+cxx11.abi
108+
2.3.110+xpu
109+
[0]: _XpuDeviceProperties(name='Intel(R) Data Center GPU Max 1550', platform_name='Intel(R) Level-Zero', type='gpu', driver_version='1.3.30049', total_memory=65536MB, max_compute_units=448, gpu_eu_count=448, gpu_subslice_count=56, max_work_group_size=1024, max_num_sub_groups=64, sub_group_sizes=[16 32], has_fp16=1, has_fp64=1, has_atomic64=1)
110+
[1]: _XpuDeviceProperties(name='Intel(R) Data Center GPU Max 1550', platform_name='Intel(R) Level-Zero', type='gpu', driver_version='1.3.30049', total_memory=65536MB, max_compute_units=448, gpu_eu_count=448, gpu_subslice_count=56, max_work_group_size=1024, max_num_sub_groups=64, sub_group_sizes=[16 32], has_fp16=1, has_fp64=1, has_atomic64=1)
90111
```
91-
2.1.0.post2+cxx11.abi
92-
2.1.30+xpu
93-
[0]: _DeviceProperties(name='Intel(R) Data Center GPU Max 1550', platform_name='Intel(R) Level-Zero', dev_type='gpu', driver_version='1.3.27642', has_fp64=1, total_memory=65536MB, max_compute_units=448, gpu_eu_count=448)
94-
[1]: _DeviceProperties(name='Intel(R) Data Center GPU Max 1550', platform_name='Intel(R) Level-Zero', dev_type='gpu', driver_version='1.3.27642', has_fp64=1, total_memory=65536MB, max_compute_units=448, gpu_eu_count=448)
95-
```
112+
96113
#### Running your own script
97114

98115
Now you are inside container with Python 3.10, PyTorch, and Intel® Extension for PyTorch\* preinstalled. You can run your own script
99116
to run on Intel GPU.
100-
101-

docker/build.sh

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,6 @@ if [[ ${IMAGE_TYPE} = "xpu" ]]; then
1212
--build-arg LEVEL_ZERO_GPU_VER=1.3.30049.10-950~22.04 \
1313
--build-arg LEVEL_ZERO_VER=1.17.6-950~22.04 \
1414
--build-arg LEVEL_ZERO_DEV_VER=1.17.6-950~22.04 \
15-
--build-arg DPCPP_VER=2024.1.0-963 \
16-
--build-arg MKL_VER=2024.1.0-691 \
17-
--build-arg CCL_VER=2021.12.0-309 \
1815
--build-arg TORCH_VERSION=2.3.1+cxx11.abi \
1916
--build-arg IPEX_VERSION=2.3.110+xpu \
2017
--build-arg TORCHVISION_VERSION=0.18.1+cxx11 \

0 commit comments

Comments
 (0)