Hello there,
There are plenty of articles in Microsoft to install CUDA on DSVM. For the latest CUDA drivers and supported operating systems, visit the NVIDIA website. Ensure that you install or upgrade to the latest supported CUDA drivers for your distribution.
Microsoft redistributes NVIDIA GRID driver installers for NV and NVv3-series VMs used as virtual workstations or for virtual applications. Install only these GRID drivers on Azure NV VMs, only on the operating systems listed in the following article.
Install NVIDIA GPU drivers on N-series VMs running Linux https://learn.microsoft.com/en-us/azure/virtual-machines/linux/n-series-driver-setup
Install NVIDIA GPU drivers on N-series VMs running Windows https://learn.microsoft.com/en-us/azure/virtual-machines/windows/n-series-driver-setup
Hope this resolves your Query !!
--If the reply is helpful, please Upvote and Accept it as an answer–