Remove tensorflow-gpu from extra requirements

Also update dependencies in requirements-tf.txt PiperOrigin-RevId: 362278786 Change-Id: Ib8943e04c36a7fe31782aecaede8cd98bfd69347
google-deepmind · Mar 11, 2021 · 76f61d5 · 76f61d5
1 parent eb06599
commit 76f61d5
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -47,7 +47,7 @@ notebook attached to a GPU or TPU.
 To get started install TensorFlow 2.0 and Sonnet 2:
 
 ```shell
-$ pip install tensorflow-gpu tensorflow-probability
+$ pip install tensorflow tensorflow-probability
 $ pip install dm-sonnet
 ```
 

diff --git a/docs/index.rst b/docs/index.rst
@@ -19,7 +19,7 @@ Installation
 
 Install Sonnet by running::
 
-    $ pip install tensorflow-gpu
+    $ pip install tensorflow
     $ pip install dm-sonnet
 
 .. toctree::

diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -1,5 +1,5 @@
 sphinx>=2.0.1
 sphinx_rtd_theme>=0.4.3
 sphinxcontrib-katex>=0.4.1
-sphinxcontrib-bibtex>=0.4.2
+sphinxcontrib-bibtex>=0.4.2,<2
 sphinx-autodoc-typehints>=1.10.3
diff --git a/requirements-test.txt b/requirements-test.txt
@@ -1,3 +1,3 @@
 mock>=3.0.5
-tensorflow-datasets>1
+tensorflow-datasets>1,<4
 docutils
diff --git a/requirements-tf.txt b/requirements-tf.txt
@@ -1 +1,2 @@
-tensorflow==2.2.1
+tensorflow==2.4.0
+tensorflow-probability==0.12.1
diff --git a/test.sh b/test.sh
@@ -28,7 +28,8 @@ virtualenv -p python3 .
 source bin/activate
 python3 --version
 
-# Run setup.py, this installs the python dependencies
+# Run setup.py, install dependencies first to use pip install
+python3 -m pip install -r requirements.txt
 python3 setup.py install
 
 # CPU count on macos or linux