diff --git a/requirements.txt b/requirements.txt index 019ea0e6..470bd0ef 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,4 +7,4 @@ sentencepiece tensorflow-text==1.15.0rc0 tensor2tensor==1.15.0 tfds-nightly -tensorflow-gpu==1.15 +tensorflow-gpu==1.15.2 diff --git a/setup.py b/setup.py index 5913371b..06258694 100644 --- a/setup.py +++ b/setup.py @@ -43,10 +43,10 @@ 'tensorflow-text==1.15.0rc0', 'tfds-nightly', 'tensor2tensor==1.15.0', - 'tensorflow-gpu==1.15', + 'tensorflow-gpu==1.15.2', ], extras_require={ - 'tensorflow': ['tensorflow-gpu==1.15'], + 'tensorflow': ['tensorflow-gpu==1.15.2'], }, classifiers=[ 'Development Status :: 4 - Beta',