Merge 04b1de2 into d987c4d

vanvalenlab · Jun 23, 2021 · 53190a7 · 53190a7
2 parents d987c4d + 04b1de2
commit 53190a7
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,7 +1,7 @@
 # Use tensorflow/tensorflow as the base image
 # Change the build arg to edit the tensorflow version.
 # Only supporting python3.
-ARG TF_VERSION=2.4.2-gpu
+ARG TF_VERSION=2.5.0-gpu
 
 FROM tensorflow/tensorflow:${TF_VERSION}
 

diff --git a/requirements.txt b/requirements.txt
@@ -3,7 +3,7 @@ pydot>=1.4.2,<2
 scipy>=1.2.3,<2
 scikit-image>=0.14.5
 scikit-learn>=0.20.4
-tensorflow~=2.4.2
+tensorflow~=2.5.0
 jupyter>=1.0.0,<2
 deepcell-tracking>=0.3.1,<0.4.0
 deepcell-toolbox>=0.9.0,<0.10.0
diff --git a/setup.py b/setup.py
@@ -62,7 +62,7 @@
         'scipy>=1.2.3,<2',
         'scikit-image>=0.14.5',
         'scikit-learn>=0.20.4',
-        'tensorflow~=2.4.2',
+        'tensorflow~=2.5.0',
         'jupyter>=1.0.0,<2',
         'opencv-python-headless<5',
         'deepcell-tracking>=0.3.1,<0.4.0',