Update CUDA and ROCm images in notebooks

Author: oksanabaza

examples/hpo-raytune/notebook/raytune-oai-demo-mlmd.ipynb

@@ -54,7 +54,7 @@
     "    min_memory=4,\n",
     "    max_memory=4,\n",
     "    num_gpus=0,\n",
-    "    image=\"quay.io/modh/ray:2.35.0-py39-cu121\",                                           \n",
+    "    image=\"quay.io/modh/ray:2.35.0-py311-cu121\",                                           \n",
     "))"
    ]
   },

examples/hpo-raytune/notebook/raytune-oai-demo.ipynb

@@ -55,7 +55,7 @@
     "    min_memory=4,\n",
     "    max_memory=4,\n",
     "    num_gpus=0,\n",
-    "    image=\"quay.io/modh/ray:2.35.0-py39-cu121\"\n",
+    "    image=\"quay.io/modh/ray:2.35.0-py311-cu121\"\n",
     "))"
    ]
   },

examples/ray-finetune-llm-deepspeed/README.md

@@ -129,7 +129,7 @@ This example has been validated on the following configurations:
         head_memory_limits=96,
         head_extended_resource_requests={'amd.com/gpu':1},
         worker_extended_resource_requests={'amd.com/gpu':1},
-        image="quay.io/rhoai/ray:2.35.0-py39-rocm61-torch24-fa26",
+        image="quay.io/rhoai/ray:2.35.0-py311-rocm61-torch24-fa26",
     )
     ```
 * Ray job:

examples/ray-finetune-llm-deepspeed/ray_finetune_llm_deepspeed.ipynb

@@ -70,11 +70,12 @@
     "    head_memory_requests=128,\n",
     "    head_memory_limits=256,\n",
     "    # Use the following parameters with NVIDIA GPUs\n",
-    "    image=\"quay.io/rhoai/ray:2.35.0-py39-cu121-torch24-fa26\",\n",
+    "    # Ensure the Python version in the notebook image matches the version used in the Ray cluster to avoid compatibility issues\n",
+    "    image=\"quay.io/rhoai/ray:2.35.0-py311-cu121-torch24-fa26\",\n",
     "    head_extended_resource_requests={'nvidia.com/gpu':1},\n",
     "    worker_extended_resource_requests={'nvidia.com/gpu':1},\n",
     "    # Or replace them with these parameters for AMD GPUs\n",
-    "    # image=\"quay.io/rhoai/ray:2.35.0-py39-rocm61-torch24-fa26\",\n",
+    "    # image=\"quay.io/rhoai/ray:2.35.0-py311-rocm61-torch24-fa26\",\n",
     "    # head_extended_resource_requests={'amd.com/gpu':1},\n",
     "    # worker_extended_resource_requests={'amd.com/gpu':1},\n",
     "))"

examples/stable-diffusion-dreambooth/notebook/00 Intro.ipynb

@@ -61,7 +61,7 @@
     "                                       worker_memory_limits=8, \n",
     "                                       head_extended_resource_requests={'nvidia.com/gpu':0},\n",
     "                                       worker_extended_resource_requests={'nvidia.com/gpu':0},\n",
-    "                                       image=\"quay.io/project-codeflare/ray:latest-py39-cu118\",\n",
+    "                                       image=\"quay.io/modh/ray:2.35.0-py311-cu121\",\n",
     "                                       write_to_file=False, # When enabled Ray Cluster yaml files are written to /HOME/.codeflare/resources \n",
     "                                       local_queue=\"local-queue\"\n",
     "                                       ))"

examples/stable-diffusion-dreambooth/notebook/raycluster.yaml

@@ -20,7 +20,7 @@ spec:
           - env:
             - name: NVIDIA_VISIBLE_DEVICES
               value: "void"
-            image: 'quay.io/project-codeflare/ray:latest-py39-cu118'
+            image: 'quay.io/modh/ray:2.35.0-py311-cu121'
             imagePullPolicy: Always
             lifecycle:
               preStop:
@@ -60,7 +60,7 @@ spec:
       template:        
         spec:
           containers:
-            - image: 'quay.io/project-codeflare/ray:latest-py39-cu118'
+            - image: 'quay.io/modh/ray:2.35.0-py311-cu121'
               lifecycle:
                 preStop:
                   exec: