Pins tf-nightly version in installation instructions (issue #74).

PiperOrigin-RevId: 364345768
Change-Id: I2dbde578913918bbea46185e2c129e7d1cbcc3de

Author: Johannes Ballé

README.md

@@ -58,7 +58,8 @@ of TF.***
 To install TFC via `pip`, run the following command:
 
 ```bash
-pip install tensorflow-compression
+pip uninstall -y tensorflow
+pip install tf-nightly==2.5.0.dev20210312 tensorflow-compression
 ```
 
 To test that the installation works correctly, you can run the unit tests with:
@@ -80,7 +81,7 @@ host. For instance, you can use a command line like this:
 
 ```bash
 docker run tensorflow/tensorflow:nightly bash -c \
-    "pip install tensorflow-compression &&
+    "pip install tf-nightly==2.5.0.dev20210312 tensorflow-compression &&
      python -m tensorflow_compression.all_tests"
 ```
 
@@ -98,7 +99,7 @@ libraries, and then installs TensorFlow and TensorFlow Compression:
 ```bash
 conda create --name ENV_NAME python=3.8 cudatoolkit=10.0 cudnn
 conda activate ENV_NAME
-pip install tensorflow-compression
+pip install tf-nightly==2.5.0.dev20210312 tensorflow-compression
 ```
 
 ## Usage

models/hific/colab.ipynb

@@ -31,7 +31,8 @@
       },
       "outputs": [],
       "source": [
-        "!pip install tensorflow-compression\n",
+        "!pip uninstall -y tensorflow\n",
+        "!pip install tf-nightly==2.5.0.dev20210312 tensorflow-compression==2.1\n",
         "![[ -e /tfc ]] || git clone https://github.com/tensorflow/compression /tfc\n",
         "%cd /tfc/models\n",
         "import tfci  # Check if tfci.py is available."

models/toy_sources/toy_sources.ipynb

@@ -37,8 +37,8 @@
         "# Run this cell to install the necessary dependencies when running the notebook\n",
         "# directly in a Colaboratory hosted runtime from Github.\n",
         "\n",
-        "%tensorflow_version 2.x\n",
-        "!pip install tensorflow-compression==2.*\n",
+        "!pip uninstall -y tensorflow\n",
+        "!pip install tf-nightly==2.5.0.dev20210312 tensorflow-compression==2.1\n",
         "![[ -e /tfc ]] || git clone https://github.com/tensorflow/compression /tfc\n",
         "%cd /tfc/models\n"
       ],