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 Notebook или локального файла сценария вы можете отправить свою модель для настройки или обучения напрямую в облако, без необходимости использования облачной консоли.
