Python 有没有可能在Google';使用可抢占tpu的s AI平台?
我最近开始使用谷歌的人工智能平台来训练我的深度神经网络模型。由于我们是一个相对较小的研究实验室,我尝试使用可抢占的TPU和主机器来训练模型。不幸的是,我在文档中没有找到一种方法来实现这一点 目前,我正在使用以下shell脚本提交培训作业:Python 有没有可能在Google';使用可抢占tpu的s AI平台?,python,tensorflow,keras,google-cloud-ai,Python,Tensorflow,Keras,Google Cloud Ai,我最近开始使用谷歌的人工智能平台来训练我的深度神经网络模型。由于我们是一个相对较小的研究实验室,我尝试使用可抢占的TPU和主机器来训练模型。不幸的是,我在文档中没有找到一种方法来实现这一点 目前,我正在使用以下shell脚本提交培训作业: !/bin/bash BUCKET_NAME="training_data" JOB_NAME="PPI_$(date +"%Y%m%d_%H%M%S")" JOB_DIR="gs://$B
!/bin/bash
BUCKET_NAME="training_data"
JOB_NAME="PPI_$(date +"%Y%m%d_%H%M%S")"
JOB_DIR="gs://$BUCKET_NAME/hp_job_dir"
TRAINER_PACKAGE_PATH="./training_job_folder/trainer"
MAIN_TRAINER_MODULE="trainer.train"
HPTUNING_CONFIG="training_job_folder/trainer/hptuning_config.yaml"
RUNTIME_VERSION=2.4
PYTHON_VERSION=3.7
REGION="us-central1"
SCALE_TIER=CUSTOM
MASTER_MACHINE_TYPE=n2-highmem-16
gcloud config set project vocal-unfolding-311510
gcloud ai-platform jobs submit training $JOB_NAME \
--job-dir $JOB_DIR \
--package-path $TRAINER_PACKAGE_PATH \
--module-name $MAIN_TRAINER_MODULE \
--region $REGION \
--runtime-version=$RUNTIME_VERSION \
--python-version=$PYTHON_VERSION \
--scale-tier $SCALE_TIER \
--config $HPTUNING_CONFIG \
--master-machine-type $MASTER_MACHINE_TYPE
gcloud ai-platform jobs stream-logs $JOB_NAME
如果有人能建议我如何将脚本更改为只使用抢占式主机或TPU,我将非常感激
提前感谢,,
曼努埃尔S