set gpu vm size

Author: Lily Pan

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

@@ -12026,6 +12026,7 @@ def test_aks_create_gpu_driver_flow(self, resource_group, resource_group_locatio
                 "node_pool_name": node_pool_name,
                 "node_pool_name_second": node_pool_name_second,
                 "ssh_key_value": self.generate_ssh_keys(),
+                "node_vm_size": "standard_nd96asr_v4"
             }
         )
 
@@ -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=[