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!
Il repository TensorFlow Cloud fornisce API che facilitano la transizione dalla creazione di modelli locali e dal debug alla formazione distribuita e all'ottimizzazione degli iperparametri su Google Cloud. Dall'interno di un Colab o Kaggle Notebook o di un file di script locale, puoi inviare il tuo modello per l'ottimizzazione o l'addestramento direttamente su Cloud, senza dover utilizzare Cloud Console.