From 11fa3de2904f3ae9c0403fce29c47be0d64fed65 Mon Sep 17 00:00:00 2001 From: Yuwen Hu <54161268+Oscilloscope98@users.noreply.github.com> Date: Fri, 24 Nov 2023 17:49:21 +0800 Subject: [PATCH] Add sutup support of win gpu for bigdl-llm (#9512) --- python/llm/setup.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/python/llm/setup.py b/python/llm/setup.py index 44ee4578..4bbd27a5 100644 --- a/python/llm/setup.py +++ b/python/llm/setup.py @@ -271,13 +271,15 @@ def setup_package(): "intel-openmp; (platform_machine=='x86_64' or platform_machine == 'AMD64')"] all_requires += CONVERT_DEP - # install with -f https://developer.intel.com/ipex-whl-stable-xpu + # Linux install with -f https://developer.intel.com/ipex-whl-stable-xpu xpu_requires = copy.deepcopy(all_requires) xpu_requires.remove('torch') - xpu_requires += ["torch==2.0.1a0", - "torchvision==0.15.2a0", + xpu_requires += ["torch==2.0.1a0;platform_system=='Linux'", + "torch==2.0.0a0+gitc6a572f;platform_system=='Windows'", + "torchvision==0.15.2a0;platform_system=='Linux'", "intel_extension_for_pytorch==2.0.110+xpu;platform_system=='Linux'", - "bigdl-core-xe==" + VERSION + ";platform_system=='Linux'", + "intel_extension_for_pytorch==2.0.110+gitba7f6c1;platform_system=='Windows'", + "bigdl-core-xe==" + VERSION + ";(platform_system=='Linux' or platform_system=='Windows')", "bigdl-core-xe-esimd==" + VERSION + ";platform_system=='Linux'"] serving_requires = ['py-cpuinfo']