问题
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
回答1:
we will support TF 2.1 officially in early Feb due to large corresponding changes on service. Thank you for your patience!
回答2:
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.
回答3:
Tensorflow 2.1 runtime version is available as of March 9th
https://cloud.google.com/ai-platform/training/docs/runtime-version-list
来源：https://stackoverflow.com/questions/59719700/tensorflow-2-on-google-cloud-ai-platform