Tensorflow 2 on Google Cloud AI platform

4
votes

Any news when Tensorflow 2 will be supported on Google Cloud AI platform?

According to the list, 1.15 is the last version to be supported: https://cloud.google.com/ml-engine/docs/runtime-version-list

3
google-cloud-ml
You can use custom-containers for Model training if needed. cloud.google.com/ai-platform/training/docs/… - gogasca

3 Answers

2
votes

we will support TF 2.1 officially in early Feb due to large corresponding changes on service. Thank you for your patience!

0
votes

Add a dependency tensorflow-gpu==2.0 to setup.py file to train with Tensorflow 2.0 in AI Platform run-time version 1.15. Tensorflow 2.1 can't use GPUs due to mismatch in CUDA libraries, but you can train with CPUs. Be careful that other dependencies don't overwrite tensorflow-gpu. Disclaimer: I trained just one job this way, and this is a really hacky solution.

0
votes

Tensorflow 2.1 runtime version is available as of March 9th

https://cloud.google.com/ai-platform/training/docs/runtime-version-list