Skip to content

Commit da3c840

Browse files
committed
min tf version
1 parent 7b42c0d commit da3c840

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

setup.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,9 @@ def get_dist(pkgname):
1212
'numpy',
1313
'regex',
1414
]
15-
if get_dist('tensorflow>=2.0.0') is None and get_dist('tensorflow_gpu>=2.0.0') is None:
16-
install_deps.append('tensorflow>=2.0.0a')
15+
tf_ver = '2.0.0a'
16+
if get_dist('tensorflow>='+tf_ver) is None and get_dist('tensorflow_gpu>='+tf_ver) is None:
17+
install_deps.append('tensorflow>='+tf_ver)
1718

1819
setup(
1920
name = 'mitdeeplearning', # How you named your package folder (MyLib)

0 commit comments

Comments
 (0)