TensorFlow Cloud è una libreria per connettere il tuo ambiente locale a 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!
Esegui in un notebook
Il repository TensorFlow Cloud fornisce API che facilitano la transizione dalla creazione di modelli locali e dal debug al training distribuito e all'ottimizzazione degli iperparametri su Google Cloud. Dall'interno di un Colab o Kaggle Notebook o da un file di script locale, puoi inviare il tuo modello per l'ottimizzazione o l'addestramento su Cloud direttamente, senza dover utilizzare Cloud Console.
