我们可以将Pytorch模型部署到Google Cloud ML Engine吗?如果是这样,怎么办?

时间:2018-07-27 07:56:40

标签: google-cloud-platform pytorch google-cloud-ml

我搜索了是否有人。有人提供了解决方法,但无法确认其是否有效。这是链接-

fastlane sources

How to setup pytorch in google-cloud-ml

当前,我正在使用Floydhub,但是开始时很好。我主要使用Google Cloud产品,因此希望使我的深度学习项目尽可能靠近。

1 个答案:

答案 0 :(得分:1)

您提到的答案可能会起作用(我也没有完全测试过)。 Cloud ML Engine团队正在努力简化这一过程。

同时,如果您更喜欢使用VM(请记住在不使用它时将其关闭),请遵循this page上的说明。例如,

export IMAGE_FAMILY="pytorch-latest-cu91"
export ZONE="us-west1-b"
export INSTANCE_NAME="my-instance"

gcloud compute instances create $INSTANCE_NAME \
  --zone=$ZONE \
  --image-family=$IMAGE_FAMILY \
  --image-project=deeplearning-platform-release \
  --maintenance-policy=TERMINATE \
  --accelerator='type=nvidia-tesla-v100,count=1' \
  --metadata='install-nvidia-driver=True'