Skip to content

Commit

Permalink
Update tf requirements
Browse files Browse the repository at this point in the history
  • Loading branch information
Irina Nicolae authored and Irina Nicolae committed Apr 11, 2018
1 parent 20a1e30 commit 48d9bf1
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 2 deletions.
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
tensorflow==1.1
tensorflow==1.5
h5py
Keras
matplotlib
Expand Down
5 changes: 4 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,8 +30,11 @@
'Keras',
'scipy',
'matplotlib',
'tensorflow',
'setuptools'],
extras_require={
"tf": ["tensorflow<=1.5.0"],
"tf_gpu": ["tensorflow-gpu<=1.5.0"],
},
# extras_require={
# 'tests': ['pytest',
# 'pytest-pep8',
Expand Down

0 comments on commit 48d9bf1

Please sign in to comment.