Skip to content

Commit 4797e76

Browse files
Cleanup tensorflow for mac arm (#2713)
CVS-161264 tensorflow-macos package was updated but tensorflow package was installed too, but shouldn't
1 parent 4600eac commit 4797e76

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

notebooks/tensorflow-quantization-aware-training/tensorflow-quantization-aware-training.ipynb

+6
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,12 @@
7272
"metadata": {},
7373
"outputs": [],
7474
"source": [
75+
"import platform\n",
76+
"\n",
77+
"\n",
78+
"if platform.system() == \"darwin\" and platform.processor() in (\"arm\", \"arm64\"):\n",
79+
" %pip uninstall -qy tensorflow\n",
80+
"\n",
7581
"%pip install -q \"openvino>=2024.6.0\" \"nncf>=2.14.0\"\n",
7682
"%pip install -q \"tensorflow-macos>=2.9.3,<2.16.0; sys_platform == 'darwin' and platform_machine == 'arm64'\"\n",
7783
"%pip install -q \"tensorflow>=2.9.3,<2.16.0; sys_platform == 'darwin' and platform_machine != 'arm64'\" # macOS x86\n",

0 commit comments

Comments
 (0)