Merge pull request #12 from valohai/bump-requirements

Bump Tensorflow patch versions
valohai · Oct 19, 2020 · 552bff5 · 552bff5
2 parents 2b4dd35 + 9ffee98
commit 552bff5
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/requirements-cpu.txt b/requirements-cpu.txt
@@ -8,4 +8,4 @@ pyparsing==2.2.0
 PyYAML==5.3
 scipy>=0.19.0
 six==1.10.0
-tensorflow==2.0.0
+tensorflow==2.0.3
diff --git a/requirements-gpu.txt b/requirements-gpu.txt
@@ -8,4 +8,4 @@ pyparsing==2.2.0
 PyYAML==5.3
 scipy>=0.19.0
 six==1.10.0
-tensorflow-gpu==2.0.0
+tensorflow-gpu==2.0.3
diff --git a/valohai.yaml b/valohai.yaml
@@ -3,7 +3,7 @@
 - step:
     name: tensorflow-check
     description: Check dependency version in the Docker image after setup and the type of hardware.
-    image: tensorflow/tensorflow:2.0.0-gpu-py3
+    image: tensorflow/tensorflow:2.0.3-gpu-py3
     command:
       - pip install --disable-pip-version-check -q -r requirements-gpu.txt
       - python --version
@@ -16,7 +16,7 @@
 - step:
     name: tensorflow-deep-dream
     description: Apply Deep Dream effect on the given image.
-    image: tensorflow/tensorflow:2.0.0-gpu-py3
+    image: tensorflow/tensorflow:2.0.3-gpu-py3
     command:
       - pip install --disable-pip-version-check -q -r requirements-gpu.txt
       - python deep_dream.py /valohai/inputs/source-image /valohai/outputs/result {parameters}
@@ -50,7 +50,7 @@
 - step:
     name: tensorflow-cifar
     description: Train a CNN model on the CIFAR10 dataset.
-    image: tensorflow/tensorflow:2.0.0-gpu-py3
+    image: tensorflow/tensorflow:2.0.3-gpu-py3
     command:
       - pip install --disable-pip-version-check -q -r requirements-gpu.txt
       - python cifar10_cnn.py {parameters}