From 4797e76daf659214d7e8b6f143ca7ef2ea85acf8 Mon Sep 17 00:00:00 2001 From: Aleksandr Mokrov Date: Tue, 4 Feb 2025 13:36:57 +0100 Subject: [PATCH] Cleanup tensorflow for mac arm (#2713) CVS-161264 tensorflow-macos package was updated but tensorflow package was installed too, but shouldn't --- .../tensorflow-quantization-aware-training.ipynb | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/notebooks/tensorflow-quantization-aware-training/tensorflow-quantization-aware-training.ipynb b/notebooks/tensorflow-quantization-aware-training/tensorflow-quantization-aware-training.ipynb index 1a6802a5257..b00796acf78 100644 --- a/notebooks/tensorflow-quantization-aware-training/tensorflow-quantization-aware-training.ipynb +++ b/notebooks/tensorflow-quantization-aware-training/tensorflow-quantization-aware-training.ipynb @@ -72,6 +72,12 @@ "metadata": {}, "outputs": [], "source": [ + "import platform\n", + "\n", + "\n", + "if platform.system() == \"darwin\" and platform.processor() in (\"arm\", \"arm64\"):\n", + " %pip uninstall -qy tensorflow\n", + "\n", "%pip install -q \"openvino>=2024.6.0\" \"nncf>=2.14.0\"\n", "%pip install -q \"tensorflow-macos>=2.9.3,<2.16.0; sys_platform == 'darwin' and platform_machine == 'arm64'\"\n", "%pip install -q \"tensorflow>=2.9.3,<2.16.0; sys_platform == 'darwin' and platform_machine != 'arm64'\" # macOS x86\n",