Skip to content

Commit 1ca21f6

Browse files
author
DEKHTIARJonathan
committed
TF 1.10.0 bug fix
1 parent 5f57e8e commit 1ca21f6

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tensorlayer/__init__.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
from __future__ import absolute_import
66

77
import os
8+
from distutils.version import LooseVersion
89

910
if 'TENSORLAYER_PACKAGE_BUILDING' not in os.environ:
1011

@@ -17,7 +18,7 @@
1718
" - `pip install --upgrade tensorflow-gpu`"
1819
)
1920

20-
if tensorflow.__version__ < "1.6.0" and os.environ.get('READTHEDOCS', None) != 'True':
21+
if LooseVersion(tensorflow.__version__) < LooseVersion("1.6.0") and os.environ.get('READTHEDOCS', None) != 'True':
2122
raise RuntimeError(
2223
"TensorLayer does not support Tensorflow version older than 1.6.0.\n"
2324
"Please update Tensorflow with:\n"
@@ -69,4 +70,5 @@
6970
__download_url__ = 'https://github.com/tensorlayer/tensorlayer'
7071
__description__ = 'Reinforcement Learning and Deep Learning Library for Researcher and Engineer.'
7172
__license__ = 'apache'
72-
__keywords__ = 'deep learning, machine learning, computer vision, nlp, supervised learning, unsupervised learning, reinforcement learning, tensorflow'
73+
__keywords__ = 'deep learning, machine learning, computer vision, nlp, '
74+
__keywords__ += 'supervised learning, unsupervised learning, reinforcement learning, tensorflow'

0 commit comments

Comments
 (0)