I'm trying to configure an Azure Batch pool to use "PyTorch from NVIDIA" VM images. I'm provisioning the pool using the Azure Python SDK and I'm getting the following error:
Reason: The specified imageReference with publisher nvidia offer pytorch_from_nvidia sku pytorch_from_nvidia_19_11_3 is not supported.
When I tried to provision the pool using the Azure Portal I was unable to find Nvidia among the supported publishers (see image below).
Is it possible to use Azure Batch with an Nvidia VM image?