Add GPU to ClusterConfiguration

project-codeflare · May 7, 2024 · 4cdb19b · 4cdb19b
1 parent a9b314e
commit 4cdb19b
Show file tree

Hide file tree

Showing 8 changed files with 9 additions and 1 deletion.
diff --git a/demo-notebooks/additional-demos/hf_interactive.ipynb b/demo-notebooks/additional-demos/hf_interactive.ipynb
@@ -92,6 +92,7 @@
     "# Create our cluster and submit\n",
     "# The SDK will try to find the name of your default local queue based on the annotation \"kueue.x-k8s.io/default-queue\": \"true\" unless you specify the local queue manually below\n",
     "cluster = Cluster(ClusterConfiguration(name='hfgputest', \n",
+    "                                       head_gpus=1, # For GPU enabled workloads set the head_gpus and num_gpus\n",
     "                                       namespace=\"default\", # Update to your namespace\n",
     "                                       num_workers=1,\n",
     "                                       min_cpus=8, \n",

diff --git a/demo-notebooks/additional-demos/local_interactive.ipynb b/demo-notebooks/additional-demos/local_interactive.ipynb
@@ -55,6 +55,7 @@
     "\n",
     "cluster = Cluster(ClusterConfiguration(namespace=namespace,\n",
     "                                       name=cluster_name,\n",
+    "                                       head_gpus=1, # For GPU enabled workloads set the head_gpus and num_gpus\n",
     "                                       num_workers=1,\n",
     "                                       min_cpus=1,\n",
     "                                       max_cpus=1,\n",

diff --git a/demo-notebooks/guided-demos/0_basic_ray.ipynb b/demo-notebooks/guided-demos/0_basic_ray.ipynb
@@ -61,7 +61,8 @@
     "# Create and configure our cluster object\n",
     "# The SDK will try to find the name of your default local queue based on the annotation \"kueue.x-k8s.io/default-queue\": \"true\" unless you specify the local queue manually below\n",
     "cluster = Cluster(ClusterConfiguration(\n",
-    "    name='raytest',\n",
+    "    name='raytest', \n",
+    "    head_gpus=1, # For GPU enabled workloads set the head_gpus and num_gpus\n",
     "    namespace='default', # Update to your namespace\n",
     "    num_workers=2,\n",
     "    min_cpus=1,\n",

diff --git a/demo-notebooks/guided-demos/2_basic_interactive.ipynb b/demo-notebooks/guided-demos/2_basic_interactive.ipynb
@@ -61,6 +61,7 @@
     "cluster_name = \"interactivetest\"\n",
     "cluster = Cluster(ClusterConfiguration(\n",
     "    name=cluster_name,\n",
+    "    head_gpus=1, # For GPU enabled workloads set the head_gpus and num_gpus\n",
     "    namespace=namespace,\n",
     "    num_workers=2,\n",
     "    min_cpus=2,\n",

diff --git a/demo-notebooks/guided-demos/notebook-ex-outputs/0_basic_ray.ipynb b/demo-notebooks/guided-demos/notebook-ex-outputs/0_basic_ray.ipynb
@@ -70,6 +70,7 @@
     "# The SDK will try to find the name of your default local queue based on the annotation \"kueue.x-k8s.io/default-queue\": \"true\" unless you specify the local queue manually below\n",
     "cluster = Cluster(ClusterConfiguration(\n",
     "    name='raytest',\n",
+    "    head_gpus=1, # For GPU enabled workloads set the head_gpus and num_gpus\n",
     "    namespace='default', # Update to your namespace\n",
     "    num_workers=2,\n",
     "    min_cpus=1,\n",

diff --git a/demo-notebooks/guided-demos/notebook-ex-outputs/2_basic_interactive.ipynb b/demo-notebooks/guided-demos/notebook-ex-outputs/2_basic_interactive.ipynb
@@ -69,6 +69,7 @@
     "cluster_name = \"interactivetest\"\n",
     "cluster = Cluster(ClusterConfiguration(\n",
     "    name=cluster_name,\n",
+    "    head_gpus=1, # For GPU enabled workloads set the head_gpus and num_gpus\n",
     "    namespace=namespace,\n",
     "    num_workers=2,\n",
     "    min_cpus=2,\n",

diff --git a/demo-notebooks/guided-demos/preview_nbs/0_basic_ray.ipynb b/demo-notebooks/guided-demos/preview_nbs/0_basic_ray.ipynb
@@ -62,6 +62,7 @@
     "# The SDK will try to find the name of your default local queue based on the annotation \"kueue.x-k8s.io/default-queue\": \"true\" unless you specify the local queue manually below\n",
     "cluster = Cluster(ClusterConfiguration(\n",
     "    name='raytest',\n",
+    "    head_gpus=1, # For GPU enabled workloads set the head_gpus and num_gpus\n",
     "    namespace='default', # Update to your namespace\n",
     "    num_workers=2,\n",
     "    min_cpus=1,\n",

diff --git a/demo-notebooks/guided-demos/preview_nbs/2_basic_interactive.ipynb b/demo-notebooks/guided-demos/preview_nbs/2_basic_interactive.ipynb
@@ -61,6 +61,7 @@
     "cluster_name = \"interactivetest\"\n",
     "cluster = Cluster(ClusterConfiguration(\n",
     "    name=cluster_name,\n",
+    "    head_gpus=1, # For GPU enabled workloads set the head_gpus and num_gpus\n",
     "    namespace=namespace,\n",
     "    num_workers=2,\n",
     "    min_cpus=2,\n",