From ea9138278c7676d86756489a307cb8135b370fad Mon Sep 17 00:00:00 2001 From: "Ma,Liangliang" Date: Mon, 21 Oct 2024 06:46:02 +0000 Subject: [PATCH] upgrade xpu-1100 CI --- .github/workflows/xpu-max1100.yml | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/.github/workflows/xpu-max1100.yml b/.github/workflows/xpu-max1100.yml index adeeb0acade2..d19e73aeef1c 100644 --- a/.github/workflows/xpu-max1100.yml +++ b/.github/workflows/xpu-max1100.yml @@ -36,7 +36,7 @@ jobs: unit-tests: runs-on: [self-hosted, intel, xpu] container: - image: intel/oneapi-basekit:2024.1.1-devel-ubuntu22.04 + image: intel/oneapi-basekit:2024.2.1-0-devel-ubuntu22.04 ports: - 80 options: --privileged -it --rm --device /dev/dri:/dev/dri -v /dev/dri/by-path:/dev/dri/by-path --ipc=host --cap-add=ALL @@ -47,12 +47,11 @@ jobs: run: | apt-get update apt-get install clinfo libaio-dev python3-pip -y - pip install torch==2.1.0.post2 -f https://developer.intel.com/ipex-whl-stable-xpu - pip install intel-extension-for-pytorch==2.1.30+xpu -f https://developer.intel.com/ipex-whl-stable-xpu - pip install intel-extension-for-pytorch-deepspeed==2.1.30 -f https://developer.intel.com/ipex-whl-stable-xpu - pip install oneccl_bind_pt==2.1.300+xpu -f https://developer.intel.com/ipex-whl-stable-xpu - pip install torchvision==0.16.0.post2 -f https://developer.intel.com/ipex-whl-stable-xpu - pip install py-cpuinfo numpy==1.26 + pip install torch==2.3.1 -f https://pytorch-extension.intel.com/release-whl/stable/xpu/us/torch/ + pip install intel-extension-for-pytorch==2.3.110+xpu -f https://pytorch-extension.intel.com/release-whl/stable/xpu/us/intel-extension-for-pytorch/ + pip install oneccl_bind_pt==2.3.100+xpu -f https://pytorch-extension.intel.com/release-whl/stable/xpu/us/oneccl-bind-pt/ + pip install torchvision==0.18.1 -f https://pytorch-extension.intel.com/release-whl/stable/xpu/us/torchvision/ + pip install py-cpuinfo numpy pip install .[dev,autotuning] - name: Check container state