From b3ef21074a29cda961909c7bda11c50b8650a730 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Thu, 2 Nov 2023 00:50:24 +0300 Subject: [PATCH] Cleanup --- installer.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/installer.py b/installer.py index 165c5b9e5..8833d8652 100644 --- a/installer.py +++ b/installer.py @@ -442,7 +442,8 @@ def check_torch(): torchvision_pip = 'https://github.com/Nuullll/intel-extension-for-pytorch/releases/download/v2.0.110%2Bxpu-master%2Bdll-bundle/torchvision-0.15.2a0+fa99a53-cp310-cp310-win_amd64.whl' ipex_pip = 'https://github.com/Nuullll/intel-extension-for-pytorch/releases/download/v2.0.110%2Bxpu-master%2Bdll-bundle/intel_extension_for_pytorch-2.0.110+gitc6ea20b-cp310-cp310-win_amd64.whl' torch_command = os.environ.get('TORCH_COMMAND', f'{pytorch_pip} {torchvision_pip} {ipex_pip}') - install('onnxruntime-openvino', ignore=True) + install('openvino', 'openvino', ignore=True) + install('onnxruntime-openvino', 'onnxruntime-openvino', ignore=True) # TODO numpy version conflicts with tensorflow and doesn't support Python 3.11 elif allow_openvino and args.use_openvino: log.info('Using OpenVINO') if "linux" in sys.platform: @@ -476,7 +477,6 @@ def check_torch(): pytorch_pip = 'torch==2.1.0' torchvision_pip = 'torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cpu' torch_command = os.environ.get('TORCH_COMMAND', f'{pytorch_pip} {torchvision_pip}') - install('onnxruntime-openvino', 'onnxruntime-openvino', ignore=True) else: machine = platform.machine() if sys.platform == 'darwin': @@ -547,6 +547,7 @@ def check_torch(): install('hidet', 'hidet') if args.use_openvino or opts.get('cuda_compile_backend', '') == 'openvino_fx': install('openvino==2023.2.0.dev20230922', 'openvino') + install('onnxruntime-openvino', 'onnxruntime-openvino', ignore=True) # TODO numpy version conflicts with tensorflow and doesn't support Python 3.11 os.environ.setdefault('PYTORCH_TRACING_MODE', 'TORCHFX') os.environ.setdefault('NEOReadDebugKeys', '1') os.environ.setdefault('ClDeviceGlobalMemSizeAvailablePercent', '100')