Thanks for reaching out to us, you can train PyTorch model in Azure Machine Learning with the latest version of Azure Machine Learning SDK v2 - https://learn.microsoft.com/en-us/azure/machine-learning/how-to-train-pytorch?view=azureml-api-2
You'll use the example scripts in this article to classify chicken and turkey images to build a deep learning neural network (DNN) based on PyTorch's transfer learning tutorial. Transfer learning is a technique that applies knowledge gained from solving one problem to a different but related problem. Transfer learning shortens the training process by requiring less data, time, and compute resources than training from scratch. To learn more about transfer learning, see the deep learning vs machine learning article.
Whether you're training a deep learning PyTorch model from the ground-up or you're bringing an existing model into the cloud, you can use Azure Machine Learning to scale out open-source training jobs using elastic cloud compute resources. You can build, deploy, version, and monitor production-grade models with Azure Machine Learning.
Another way you may want to try is DSVM, please refer to this repo -
https://github.com/leestott/Pytorch-Azure-Machine-Learning-Services/blob/master/Setup.md
Please take a look at those docs, I hope it helps.
Regards,
Yutong
-Please kindly accept the answer if you feel helpful to support the community, thanks a lot.