Skip to content

Upgrade Tensorflow version #2079

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Jan 11, 2023
Merged

Upgrade Tensorflow version #2079

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 1 addition & 2 deletions cmd/metricscollector/v1beta1/tfevent-metricscollector/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,5 +2,4 @@ psutil==5.9.4
rfc3339>=6.2
grpcio==1.41.1
googleapis-common-protos==1.6.0
tensorflow==2.9.3; platform_machine=="x86_64"
tensorflow-aarch64==2.9.1; platform_machine=="aarch64"
tensorflow==2.11.0
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Since Tensorflow v2.10.0, the binary for aarch64 is shipped as tensorflow.

aarch64 CPUs: Experimental performance optimizations from Compute Library for the Arm® Architecture (ACL) are available through oneDNN in the default Linux aarch64 package (pip install tensorflow).

https://github.com/tensorflow/tensorflow/releases/tag/v2.10.0

3 changes: 1 addition & 2 deletions cmd/suggestion/nas/enas/v1beta1/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
grpcio==1.41.1
googleapis-common-protos==1.6.0
cython>=0.29.24
tensorflow==2.9.3; platform_machine=="x86_64"
tensorflow-aarch64==2.9.1; platform_machine=="aarch64"
tensorflow==2.11.0
7 changes: 4 additions & 3 deletions examples/v1beta1/trial-images/enas-cnn-cifar10/Dockerfile.gpu
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
# tensorflow-gpu=2.9.1, cuda=11.7.1
# Ref: https://docs.nvidia.com/deeplearning/frameworks/tensorflow-release-notes/rel_22-08.html#rel_22-08
FROM nvcr.io/nvidia/tensorflow:22.08-tf2-py3
# We need to use the nvcr.io/nvidia/tensorflow image as a base image to support both linux/amd64 and linux_arm64 platforms.
# tensorflow-gpu=2.10.1, cuda=11.8.0
# Ref: https://docs.nvidia.com/deeplearning/frameworks/tensorflow-release-notes/rel-22-12.html#rel-22-12
FROM nvcr.io/nvidia/tensorflow:22.12-tf2-py3

ENV TARGET_DIR /opt/enas-cnn-cifar10
ENV PYTHONPATH ${TARGET_DIR}
Expand Down
2 changes: 1 addition & 1 deletion examples/v1beta1/trial-images/enas-cnn-cifar10/RunTrial.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@
test_model = constructor.build_model()
test_model.summary()
test_model.compile(loss=keras.losses.categorical_crossentropy,
optimizer=keras.optimizers.Adam(learning_rate=1e-3, decay=1e-4),
optimizer=keras.optimizers.Adam(learning_rate=1e-3),
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

To resolve the following error:

Traceback (most recent call last):
  File "/opt/enas-cnn-cifar10/RunTrial.py", line 66, in <module>
    optimizer=keras.optimizers.Adam(learning_rate=1e-3, decay=1e-4),
  File "/usr/local/lib/python3.10/site-packages/keras/optimizers/optimizer_experimental/adam.py", line 104, in __init__
    super().__init__(
  File "/usr/local/lib/python3.10/site-packages/keras/optimizers/optimizer_experimental/optimizer.py", line 1053, in __init__
    super().__init__(
  File "/usr/local/lib/python3.10/site-packages/keras/optimizers/optimizer_experimental/optimizer.py", line 89, in __init__
    self._process_kwargs(kwargs)
  File "/usr/local/lib/python3.10/site-packages/keras/optimizers/optimizer_experimental/optimizer.py", line 118, in _process_kwargs
    raise ValueError(
ValueError: decay is deprecated in the new Keras optimizer, pleasecheck the docstring for valid arguments, or use the legacy optimizer, e.g., tf.keras.optimizers.legacy.Adam.

The decay for the learning rate was removed from the Adam optimizer.

metrics=['accuracy'])

print(">>> Model Constructed Successfully\n")
Expand Down
3 changes: 1 addition & 2 deletions examples/v1beta1/trial-images/enas-cnn-cifar10/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,3 +1,2 @@
scipy>=1.7.2
tensorflow==2.9.3; platform_machine=="x86_64"
tensorflow-aarch64==2.9.1; platform_machine=="aarch64"
tensorflow==2.11.0
3 changes: 1 addition & 2 deletions examples/v1beta1/trial-images/tf-mnist-with-summaries/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,2 +1 @@
tensorflow==2.9.3; platform_machine=="x86_64"
tensorflow-aarch64==2.9.1; platform_machine=="aarch64"
tensorflow==2.11.0
4 changes: 2 additions & 2 deletions sdk/python/v1beta1/kubeflow/katib/constants/constants.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -32,8 +32,8 @@

# Supported base images for the Katib Trials.
# TODO (andreyvelich): Implement list_base_images function to get each image description.
BASE_IMAGE_TENSORFLOW = "docker.io/tensorflow/tensorflow:2.9.1"
BASE_IMAGE_TENSORFLOW_GPU = "docker.io/tensorflow/tensorflow:2.9.1-gpu"
BASE_IMAGE_TENSORFLOW = "docker.io/tensorflow/tensorflow:2.11.0"
BASE_IMAGE_TENSORFLOW_GPU = "docker.io/tensorflow/tensorflow:2.11.0-gpu"
BASE_IMAGE_PYTORCH = "docker.io/pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime"
BASE_IMAGE_MXNET = "docker.io/mxnet/python:1.9.1_native_py3"

Expand Down