diff --git a/mlvm/test_mlvm.py b/mlvm/test_mlvm.py
index 121bec6d0..004545fc0 100644
--- a/mlvm/test_mlvm.py
+++ b/mlvm/test_mlvm.py
@@ -22,6 +22,7 @@ class MLVMTestCase(DataprocTestCase):
   RAPIDS_DASK_SCRIPT = "verify_rapids_dask.py"
   DASK_YARN_SCRIPT = "verify_dask_yarn.py"
   DASK_STANDALONE_SCRIPT = "verify_dask_standalone.py"
+  #dummy
 
   def verify_python(self):
     self.assert_dataproc_job(
@@ -123,11 +124,8 @@ def test_mlvm_gpu(self, configuration, dask_runtime, rapids_runtime):
     if self.getImageVersion() < pkg_resources.parse_version("2.0"):
       self.skipTest("Not supported in pre 2.0 images")
 
-    metadata = ("init-actions-repo={},include-gpus=true"
-                ",gpu-driver-provider=NVIDIA").format(self.INIT_ACTIONS_REPO)
-
-    cudnn_version = "8.1.1.33"
-    cuda_version = "11.2"
+    cudnn_version = "9.1.0.70"
+    cuda_version = "12.4"
 
     metadata = ("init-actions-repo={},include-gpus=true"
                 ",gpu-driver-provider=NVIDIA,"
@@ -143,7 +141,7 @@ def test_mlvm_gpu(self, configuration, dask_runtime, rapids_runtime):
         configuration,
         self.INIT_ACTIONS,
         optional_components=self.OPTIONAL_COMPONENTS,
-        machine_type="n1-standard-4",
+        machine_type="n1-highmem-8",
         master_accelerator="type=nvidia-tesla-t4",
         worker_accelerator="type=nvidia-tesla-t4",
         timeout_in_minutes=60,