PyTorch on Google Cloud: How to train PyTorch models on AI Platform

Click for: original source

PyTorch has been predominantly used in research and in recent years it has gained tremendous traction in the industry as well due to its ease of use and deployment. By Rajesh Thallam, Vaibhav Singh.

Google Cloud AI Platform is a fully managed end-to-end platform for data science and machine learning on Google Cloud. Leveraging Google’s expertise in AI, AI Platform offers a flexible, scalable and reliable platform to run your machine learning workloads. AI Platform has built-in support for PyTorch through Deep Learning Containers that are performance optimized, compatibility tested and ready to deploy.

Authors then explain:

  • Why PyTorch on Google Cloud AI platform?
  • Use case and dataset
  • Creating a development environment on AI Platform Notebooks
  • Creating a Notebook instance with the pre-built PyTorch DLVM image
  • Notebook instance with custom container
  • Training a PyTorch model on AI Platform training

You will find plenty opf resources for further study in this article. Authors explored Cloud AI Platform Notebooks as a fully customizable IDE for PyTorch model development. We then trained the model on Cloud AI Platform Training service, a fully managed service for training machine learning models at scale. Nice one!

[Read More]

Tags gcp google data-science python json