Fix OpenVINO Torchvision install on ARM

This commit is contained in:
Disty0
2025-08-07 16:23:22 +03:00
parent 31e6cfd91b
commit 4875c26de3
+1 -1
View File
@@ -818,7 +818,7 @@ def install_openvino():
if sys.platform == 'darwin':
torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.8.0 torchvision==0.23.0')
else:
torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.8.0+cpu torchvision==0.23.0+cpu --index-url https://download.pytorch.org/whl/cpu')
torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.8.0+cpu torchvision==0.23.0 --index-url https://download.pytorch.org/whl/cpu')
install(os.environ.get('OPENVINO_COMMAND', 'openvino==2025.2.0'), 'openvino')
install(os.environ.get('NNCF_COMMAND', 'nncf==2.17.0'), 'nncf')