1

为了获得在线预测，我正在 ai 平台上创建模型版本。除非我想使用 --accelerator 参数，否则它工作正常。

这是有效的命令：

    gcloud alpha ai-platform versions create [...] --model [...] --origin=[...] --python-version=3.5 --runtime-version=1.14 --package-uris=[...] --machine-type=mls1-c4-m4 --prediction-class=[...]

这是使它不起作用的参数：

    --accelerator=^:^count=1:type=nvidia-tesla-k80

这是我收到的错误消息：

    ERROR: (gcloud.alpha.ai-platform.versions.create) INVALID_ARGUMENT: Request contains an invalid argument.

我希望它能够工作，因为 1）参数存在并使用这两个键（计数和类型），2）我对参数使用正确的语法，任何其他语法都会返回语法错误，以及 3）“nvidia- tesla-k80" 值存在（在 --help 中描述）并且在部署模型的区域中可用。

4

2 回答 2

2

确保您使用的是最新版本的Google Cloud SDK。

然后您可以使用以下命令：

gcloud beta ai-platform versions create $VERSION_NAME \
  --model $MODEL_NAME \
  --origin gs://$MODEL_DIRECTORY_URI \
  --runtime-version 1.15 \
  --python-version 3.7 \
  --framework tensorflow \
  --machine-type n1-standard-4 \
  --accelerator count=1,type=nvidia-tesla-t4

作为参考，您可以在模型创建期间启用日志记录：

gcloud beta ai-platform models create {MODEL_NAME} \
 --regions {REGION} 
 --enable-logging \
 --enable-console-logging
于 2020-02-20T05:01:17.930 回答
0

您可以在官方文档中查看 --accelerator 参数的格式为：

--accelerator=count=1,type=nvidia-tesla-k80

我认为这可能会导致您的问题，请告诉我。

于 2019-10-07T10:24:00.613 回答