使用Google Cloud AI平台和GPU运行自定义预测
使用Google Cloud AI平台和GPU运行自定义预测
提问于 2019-11-14 14:23:07
我正在尝试将MaskRCNN模型部署到GCP平台。默认的在线预测对我不起作用，因为我的输入大小超过了允许的限制。因此，我决定继续使用自定义预测路线。
关于自定义预测，我尝试使用比默认提供的机器类型更大的机器类型，因为我的模型很大，并且需要比默认模型( mls1-c1-m2 )更多的内存。
我想部署n1-standard-4机器类型的模型，下面是我正在运行的命令：
gcloud beta ai-platform versions create $VERSION_NAME --model $MODEL_NAME --runtime-version 1.14 --python-version 3.5 --origin $MODEL_DIR --package-uris gs://pred_02/staging/my_custom_code-0.1.tar.gz --framework tensorflow --prediction-class predictor.MyPredictor --machine-type n1-standard-4 但这会抛出一个错误：
ERROR: (gcloud.beta.ai-platform.versions.create) FAILED_PRECONDITION: Machine type n1-standard-4 only support FRAMEWORK_TENSORFLOW我也尝试了--framework arg的不同变体，但它们都抛出了一些类似的错误。那么，它为什么抛出这个错误呢？
如果这是可行的，那么我的想法是将这种机器类型与k80加速器一起使用。这是我的第二个问题吗？我可以使用加速器运行它来进行自定义预测吗？
回答 1
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/58850577
复制相关文章
相似问题
腾讯云开发者
