Running TAO in Microsoft Azure

[ahaselhan]

Hello, I am hoping to get some assistance and clarification about setting up an Azure VM to run the TAO Toolkit. I have been reading the guide provided by NVIDIA found here: Running TAO on an Azure VM

However, I think this may be a little out of date… On the guide it shows setting up using NVIDIA GPU-Optimized Image for AI and HPC

But I think the names have changed since this was posted. Is the correct image to select the NVIDIA GPU-Optimized VMI?

If so, I was looking at the plans, and noticed that it used Ubuntu 22.04

But I see that TAO requires Ubuntu 20.04 based on what I see here: https://docs.nvidia.com/tao/tao-toolkit-archive/5.2.0/text/tao_toolkit_quick_start_guide.html

[Morganh]

@ahaselhan

But I see that TAO requires Ubuntu 20.04 based on what I see here: TAO Toolkit Quick Start Guide - NVIDIA Docs

For latest 5.5.0 version(NVIDIA TAO - NVIDIA Docs → Getting Started - NVIDIA Docs), this page mentions that it uses Ubuntu22.04 now.

[ahaselhan]

Can you also confirm the image NVIDIA GPU-Optimized VMI is the correct image if I am trying to follow the step to run TAO on an Azure VM? The image shown by the setup guide appears to have been changed/renamed.

Thanks,
Andrew

[Morganh]

Yes, it is " NVIDIA GPU-Optimized VMI" . The link is: GPU optimized image.