set gpu vm size

Author: Lily Pan

src/azure-cli/azure/cli/command_modules/acs/tests/latest/test_aks_commands.py

@@ -12016,6 +12016,7 @@ def test_aks_create_gpu_driver_flow(self, resource_group, resource_group_locatio
         aks_name = self.create_random_name("cliakstest", 16)
         node_pool_name = self.create_random_name("c", 6)
         node_pool_name_second = self.create_random_name("c", 6)
+        node_vm_size = "Standard_NC6s_v3"
         self.kwargs.update(
             {
                 "resource_group": resource_group,
@@ -12045,6 +12046,7 @@ def test_aks_create_gpu_driver_flow(self, resource_group, resource_group_locatio
             "--resource-group={resource_group} "
             "--cluster-name={name} "
             "--name={node_pool_name} "
+            "--node-vm-size={node_vm_size} "
             "--os-type Linux "
             "--gpu-driver None",
             checks=[
@@ -12059,6 +12061,7 @@ def test_aks_create_gpu_driver_flow(self, resource_group, resource_group_locatio
             "--resource-group={resource_group} "
             "--cluster-name={name} "
             "--name={node_pool_name_second} "
+            "--node-vm-size={node_vm_size} "
             "--os-type Linux "
             "--gpu-driver Install",
             checks=[