Merge pull request #4994 from RasaHQ/use-tf-cpu-only

use tf without gpu support by default
RasaHQ · Jan 16, 2020 · 7c14020 · 7c14020
2 parents ef93c75 + 916dc11
commit 7c14020
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/changelog/4989.improvement.rst b/changelog/4989.improvement.rst
@@ -0,0 +1,3 @@
+Switching back to a TensorFlow release which only includes CPU support to reduce the
+size of the dependencies. If you want to use the TensorFlow package with GPU support,
+please run ``tensorflow-gpu==1.15.0``.
diff --git a/requirements.txt b/requirements.txt
@@ -8,7 +8,7 @@ redis==3.3.5
 pymongo[tls,srv]==3.8.0
 numpy==1.16.3
 scipy==1.2.1
-tensorflow==1.15.0
+tensorflow-cpu==1.15.0
 absl-py>=0.8.0
 # setuptools comes from tensorboard requirement:
 # https://github.com/tensorflow/tensorboard/blob/1.14/tensorboard/pip_package/setup.py#L33

diff --git a/setup.py b/setup.py
@@ -37,7 +37,7 @@
     "pymongo[tls,srv]~=3.8",
     "numpy~=1.16",
     "scipy~=1.2",
-    "tensorflow~=1.15.0",
+    "tensorflow-cpu~=1.15.0",
     # absl is a tensorflow dependency, but produces double logging before 0.8
     # should be removed once tensorflow requires absl > 0.8 on its own
     "absl-py>=0.8.0",