A100 can't use all GPU instance profiles in MIG

Hi,

When trying to setup the MIG instance on a A100 I run into an issue where profiles 5 and 0 are unusable.
Whenever I try to select these profiles the machine just hangs and starts to print error messages. Only way to recover is to reboot the machine.

nmi watchdog: BUG: soft lockup - CPU#6 stuck for 23s! [VM Thread:4054]

image

Any help is welcome
Thanks