From 127c71a22a8a0e9257162640a13e953bde270641 Mon Sep 17 00:00:00 2001 From: Song Fuchang Date: Fri, 15 Dec 2023 14:19:01 +0800 Subject: [PATCH] Update IPEX to 2.1.10+xpu (#4931) * This will require Intel oneAPI Toolkit 2024.0 --- one_click.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/one_click.py b/one_click.py index 7e2cf7e8..76e8580e 100644 --- a/one_click.py +++ b/one_click.py @@ -218,7 +218,7 @@ def install_webui(): elif is_linux() and (choice == "C" or choice == "N"): install_pytorch = "python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu" elif choice == "D": - install_pytorch = "python -m pip install torch==2.0.1a0 torchvision==0.15.2a0 intel_extension_for_pytorch==2.0.110+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/" + install_pytorch = "python -m pip install torch==2.1.0a0 torchvision==0.16.0a0 intel_extension_for_pytorch==2.1.10+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/" # Install Git and then Pytorch print_big_message("Installing PyTorch.")