O TensorFlow Cloud é uma biblioteca para conectar seu ambiente local ao Google Cloud.

import tensorflow_cloud as tfc

TF_GPU_IMAGE = "tensorflow/tensorflow:latest-gpu"

run_parameters = {
  distribution_strategy='auto',
  requirements_txt='requirements.txt',
  docker_config=tfc.DockerConfig(
    parent_image=TF_GPU_IMAGE,
    image_build_bucket=GCS_BUCKET
  ),
  chief_config=tfc.COMMON_MACHINE_CONFIGS['K80_1X'],
  worker_config=tfc.COMMON_MACHINE_CONFIGS['K80_1X'],
  worker_count=3,
  job_labels={'job': "my_job"}
}

tfc.run(**run_parameters) # Runs your training on Google Cloud!
Execute em um notebook
O repositório TensorFlow Cloud fornece APIs que facilitam a transição da construção e depuração de modelos locais para treinamento distribuído e ajuste de hiperparâmetros no Google Cloud. De dentro de um Colab ou Kaggle Notebook ou de um arquivo de script local, você pode enviar seu modelo para ajuste ou treinamento diretamente na nuvem, sem precisar usar o Cloud Console.