我想在Google AI平台计算机上部署我的Tensorflow模型。我已经在旧版计算机mls1-c4-m2上有一个可用的版本。现在,我想使预测更快,并使用带有K80 Nvidia GPU的n1-standard-4机器。
我正在遵循本指南:https://cloud.google.com/ai-platform/prediction/docs/machine-types-online-prediction
如指南中所述,我已在us-central1中实例化了一个模型。
这是命令:
gcloud beta ai-platform versions create $VERSION \
--model $MODEL \
--origin $ORIGIN \
--runtime-version 1.15 \
--python-version 3.7 \
--framework tensorflow \
--machine-type n1-standard-4 \
--accelerator count=1,type=nvidia-tesla-k80 \
--config $CONFIG
这是配置文件:
runtimeVersion: '1.15'
manualScaling:
nodes: 1
这是我得到的错误:
ERROR: (gcloud.beta.ai-platform.versions.create) unrecognized arguments:
--accelerator
count=1,type=nvidia-tesla-k80
有人经历过吗?
似乎无法识别--accelerator标志。也许您需要升级gcloud组件