ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Add sutup support of win gpu for bigdl-llm (#9512)

Browse source
This commit is contained in:
Yuwen Hu 2023-11-24 17:49:21 +08:00 committed by GitHub
parent 45820cf3b9
commit 11fa3de290
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
python/llm/setup.py
View file

@ -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']