Currently AI Platform training only provides pre-built Pytorch…
Submit a training model job in Google Cloud Datalab