Question

为了获得在线预测，我在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中进行了描述），并且在部署模型的区域中可用。

Answer 1

确保您使用的是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

Answer 2

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

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

我想这可能会导致您的问题。

Google Cloud AI平台：我无法使用“ --accelerator”参数创建模型版本

2 个答案: