Fix tf 2.14 installation in tutorials (#1346)

Add patch part to TF version in tutorials to make sure TF 2.14 is installed (without mentioning the patch version and using ~= in pip install, it kept the 2.17 version that is installed in the colab). In addition, remove [and-cuda] to enable successful installation in colab. Co-authored-by: reuvenp <[email protected]>
sony · Jan 28, 2025 · c3ec981 · c3ec981
1 parent 1b2548e
commit c3ec981
Show file tree

Hide file tree

Showing 9 changed files with 14 additions and 14 deletions.
diff --git a/...ls/notebooks/mct_features_notebooks/keras/example_keras_activation_threshold_search.ipynb b/...ls/notebooks/mct_features_notebooks/keras/example_keras_activation_threshold_search.ipynb
@@ -74,8 +74,8 @@
    },
    "outputs": [],
    "source": [
-    "TF_VER = '2.14'\n",
-    "!pip install -q tensorflow[and-cuda]~={TF_VER}"
+    "TF_VER = '2.14.0'\n",
+    "!pip install -q tensorflow~={TF_VER}"
    ]
   },
   {

diff --git a/...s/notebooks/mct_features_notebooks/keras/example_keras_activation_z_score_threshold.ipynb b/...s/notebooks/mct_features_notebooks/keras/example_keras_activation_z_score_threshold.ipynb
@@ -58,8 +58,8 @@
    },
    "outputs": [],
    "source": [
-    "TF_VER = '2.14'\n",
-    "!pip install -q tensorflow[and-cuda]~={TF_VER}"
+    "TF_VER = '2.14.0'\n",
+    "!pip install -q tensorflow~={TF_VER}"
    ]
   },
   {

diff --git a/tutorials/notebooks/mct_features_notebooks/keras/example_keras_mobilenet_gptq.ipynb b/tutorials/notebooks/mct_features_notebooks/keras/example_keras_mobilenet_gptq.ipynb
@@ -44,7 +44,7 @@
    "source": [
     "TF_VER = '2.14.0'\n",
     "\n",
-    "!pip install -q tensorflow[and-cuda]~={TF_VER}"
+    "!pip install -q tensorflow~={TF_VER}"
    ]
   },
   {

diff --git a/...ials/notebooks/mct_features_notebooks/keras/example_keras_mobilenet_mixed_precision.ipynb b/...ials/notebooks/mct_features_notebooks/keras/example_keras_mobilenet_mixed_precision.ipynb
@@ -39,8 +39,8 @@
    },
    "outputs": [],
    "source": [
-    "TF_VER = '2.14'\n",
-    "!pip install -q tensorflow[and-cuda]~={TF_VER}"
+    "TF_VER = '2.14.0'\n",
+    "!pip install -q tensorflow~={TF_VER}"
    ]
   },
   {

diff --git a/tutorials/notebooks/mct_features_notebooks/keras/example_keras_network_editor.ipynb b/tutorials/notebooks/mct_features_notebooks/keras/example_keras_network_editor.ipynb
@@ -47,8 +47,8 @@
    },
    "outputs": [],
    "source": [
-    "TF_VER = '2.14'\n",
-    "!pip install -q tensorflow[and-cuda]~={TF_VER}"
+    "TF_VER = '2.14.0'\n",
+    "!pip install -q tensorflow~={TF_VER}"
    ]
   },
   {

diff --git a/...als/notebooks/mct_features_notebooks/keras/example_keras_post-training_quantization.ipynb b/...als/notebooks/mct_features_notebooks/keras/example_keras_post-training_quantization.ipynb
@@ -31,8 +31,8 @@
    "execution_count": null,
    "outputs": [],
    "source": [
-    "TF_VER = '2.14'\n",
-    "!pip install -q tensorflow[and-cuda]~={TF_VER}"
+    "TF_VER = '2.14.0'\n",
+    "!pip install -q tensorflow~={TF_VER}"
    ],
    "metadata": {
     "collapsed": false

diff --git a/tutorials/notebooks/mct_features_notebooks/keras/example_keras_pruning_mnist.ipynb b/tutorials/notebooks/mct_features_notebooks/keras/example_keras_pruning_mnist.ipynb
@@ -44,7 +44,7 @@
    "cell_type": "code",
    "source": [
     "TF_VER = '2.14.0'\n",
-    "!pip install -q tensorflow[and-cuda]~={TF_VER}"
+    "!pip install -q tensorflow~={TF_VER}"
    ],
    "metadata": {
     "id": "xTvVA__4NItc"

diff --git a/tutorials/notebooks/mct_features_notebooks/keras/example_keras_qat.ipynb b/tutorials/notebooks/mct_features_notebooks/keras/example_keras_qat.ipynb
@@ -33,7 +33,7 @@
    "outputs": [],
    "source": [
     "TF_VER = '2.14.0'\n",
-    "!pip install -q tensorflow[and-cuda]~={TF_VER}"
+    "!pip install -q tensorflow~={TF_VER}"
    ]
   },
   {

diff --git a/tutorials/notebooks/mct_features_notebooks/keras/example_keras_xquant.ipynb b/tutorials/notebooks/mct_features_notebooks/keras/example_keras_xquant.ipynb
@@ -41,7 +41,7 @@
    "source": [
     "TF_VER = '2.14.0'\n",
     "\n",
-    "!pip install -q tensorflow[and-cuda]~={TF_VER}"
+    "!pip install -q tensorflow~={TF_VER}"
    ],
    "metadata": {
     "id": "kCLHJUhTlPDi"