Skip to content

Commit 7c14020

Browse files
authored
Merge pull request #4994 from RasaHQ/use-tf-cpu-only
use tf without gpu support by default
2 parents ef93c75 + 916dc11 commit 7c14020

File tree

3 files changed

+5
-2
lines changed

3 files changed

+5
-2
lines changed

changelog/4989.improvement.rst

+3
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
Switching back to a TensorFlow release which only includes CPU support to reduce the
2+
size of the dependencies. If you want to use the TensorFlow package with GPU support,
3+
please run ``tensorflow-gpu==1.15.0``.

requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ redis==3.3.5
88
pymongo[tls,srv]==3.8.0
99
numpy==1.16.3
1010
scipy==1.2.1
11-
tensorflow==1.15.0
11+
tensorflow-cpu==1.15.0
1212
absl-py>=0.8.0
1313
# setuptools comes from tensorboard requirement:
1414
# https://github.com/tensorflow/tensorboard/blob/1.14/tensorboard/pip_package/setup.py#L33

setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@
3737
"pymongo[tls,srv]~=3.8",
3838
"numpy~=1.16",
3939
"scipy~=1.2",
40-
"tensorflow~=1.15.0",
40+
"tensorflow-cpu~=1.15.0",
4141
# absl is a tensorflow dependency, but produces double logging before 0.8
4242
# should be removed once tensorflow requires absl > 0.8 on its own
4343
"absl-py>=0.8.0",

0 commit comments

Comments
 (0)