TensorFlow Cloud 資料庫可將您的本機環境連結到 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"}
]

run(**run_parameters) # Runs your training on Google Cloud!
TensorFlow Cloud 存放區提供的 API 能讓您輕鬆從在本機建構模型和偵錯,改為在 Google Cloud 上進行分散式訓練和超參數微調。您可從 Colab、Kaggle 筆記本或本機指令碼檔案傳送要微調的模型,或直接在 Cloud 上訓練模型,不需要使用 Cloud Console。