diff --git a/README.md b/README.md
index f996ffd..2494f5c 100644
--- a/README.md
+++ b/README.md
@@ -22,6 +22,11 @@ or directly from the repository as
 pip install "git+https://github.com/google-research/robustness_metrics.git#egg=robustness_metrics"
 ```
 
+Note that `robustness_metrics` requires TensorFlow and TensorFlow Probability,
+but installing `robustness_metrics` does not automatically install them. You
+will need to install TensorFlow ( `tensorflow` or `tf-nightly`) and TensorFlow
+Probability (`tensorflow_probability` or `tfp-nightly`).
+
 There are three steps to evaluate a model: 1. import the model; 2. launch an
 experiment; and 3. examine results.
 
diff --git a/setup.py b/setup.py
index 6619e11..22ca6a6 100644
--- a/setup.py
+++ b/setup.py
@@ -34,10 +34,13 @@
         "tabulate",
         "tensorflow_datasets",
         "tensorflow_hub",
-        "tf-nightly",
-        "tfp-nightly",
     ],
-    extras_require={},
+    extras_require={
+        "tensorflow": [
+            "tf-nightly",
+            "tfp-nightly",
+        ],
+    },
     classifiers=[
         "Development Status :: 4 - Beta",
         "Intended Audience :: Developers",