How to create a VM image with the NVMe support

We build custom VM images and publish them to Azure Compute Gallery. And now we are trying to build images for NVMe-only VM series such as Dsv6. We've built an image based on Ubuntu 22.04 (0001-com-ubuntu-server-jammy/22_04-lts-gen2) on Standard_D2s_v6, but creating a new Standard_D2s_v6 VM from that image fails with:

'Standard_D2s_v6' cannot boot with OS image 

                    or disk. Please check that disk controller  

                    types supported by the OS image or disk is  

                    one of the supported disk controller types  

                    for the VM size 'Standard_D2s_v6'.

If I set disk_controller_type="NVMe" when creating a VM, I get a different error:

                    Message: Disk controller type 'NVMe' not    
                    supported for user VM image.                
                    Target: storageProfile.diskControllerType 

Apparently, the VM image must be marked as supporting NVMe explicitly according to the docs, but I can't find how to do that. Can you provide instructions how to mark a VM image as supporting NVMe so that it works with the Dsv6 series.