diff --git a/notebooks/AlphaFold.ipynb b/notebooks/AlphaFold.ipynb
index 2ce4ccf44b795595656c5d2e251a371112d8599b..6736838c0c5b19b08c7545c678d48e392f4833ab 100644
--- a/notebooks/AlphaFold.ipynb
+++ b/notebooks/AlphaFold.ipynb
@@ -62,6 +62,9 @@
     "try:\n",
     "  with tqdm.notebook.tqdm(total=100, bar_format=TQDM_BAR_FORMAT) as pbar:\n",
     "    with io.capture_output() as captured:\n",
+    "      # Uninstall default Colab version of TF.\n",
+    "      %shell pip uninstall -y tensorflow\n",
+    "\n",
     "      %shell sudo apt install --quiet --yes hmmer\n",
     "      pbar.update(6)\n",
     "\n",
diff --git a/setup.py b/setup.py
index 762bd7be8f03bdb6142493dcd8a419dcd066939d..0006a2aa76a7253fe2facc720d11c6c54c581acf 100644
--- a/setup.py
+++ b/setup.py
@@ -39,7 +39,7 @@ setup(
         'ml-collections',
         'numpy',
         'scipy',
-        'tensorflow',
+        'tensorflow-cpu',
     ],
     tests_require=['mock'],
     classifiers=[