Skip to content

Commit c7ff66b

Browse files
committed
Add GPU to ClusterConfiguration
1 parent a9b314e commit c7ff66b

File tree

6 files changed

+7
-1
lines changed

6 files changed

+7
-1
lines changed

demo-notebooks/guided-demos/0_basic_ray.ipynb

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,8 @@
6161
"# Create and configure our cluster object\n",
6262
"# 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",
6363
"cluster = Cluster(ClusterConfiguration(\n",
64-
" name='raytest',\n",
64+
" name='raytest', \n",
65+
" head_gpus=1, # For GPU enabled workloads set the head_gpus and num_gpus\n",
6566
" namespace='default', # Update to your namespace\n",
6667
" num_workers=2,\n",
6768
" min_cpus=1,\n",

demo-notebooks/guided-demos/2_basic_interactive.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@
6161
"cluster_name = \"interactivetest\"\n",
6262
"cluster = Cluster(ClusterConfiguration(\n",
6363
" name=cluster_name,\n",
64+
" head_gpus=1, # For GPU enabled workloads set the head_gpus and num_gpus\n",
6465
" namespace=namespace,\n",
6566
" num_workers=2,\n",
6667
" min_cpus=2,\n",

demo-notebooks/guided-demos/notebook-ex-outputs/0_basic_ray.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,7 @@
7070
"# 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",
7171
"cluster = Cluster(ClusterConfiguration(\n",
7272
" name='raytest',\n",
73+
" head_gpus=1, # For GPU enabled workloads set the head_gpus and num_gpus\n",
7374
" namespace='default', # Update to your namespace\n",
7475
" num_workers=2,\n",
7576
" min_cpus=1,\n",

demo-notebooks/guided-demos/notebook-ex-outputs/2_basic_interactive.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@
6969
"cluster_name = \"interactivetest\"\n",
7070
"cluster = Cluster(ClusterConfiguration(\n",
7171
" name=cluster_name,\n",
72+
" head_gpus=1, # For GPU enabled workloads set the head_gpus and num_gpus\n",
7273
" namespace=namespace,\n",
7374
" num_workers=2,\n",
7475
" min_cpus=2,\n",

demo-notebooks/guided-demos/preview_nbs/0_basic_ray.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@
6262
"# 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",
6363
"cluster = Cluster(ClusterConfiguration(\n",
6464
" name='raytest',\n",
65+
" head_gpus=1, # For GPU enabled workloads set the head_gpus and num_gpus\n",
6566
" namespace='default', # Update to your namespace\n",
6667
" num_workers=2,\n",
6768
" min_cpus=1,\n",

demo-notebooks/guided-demos/preview_nbs/2_basic_interactive.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@
6161
"cluster_name = \"interactivetest\"\n",
6262
"cluster = Cluster(ClusterConfiguration(\n",
6363
" name=cluster_name,\n",
64+
" head_gpus=1, # For GPU enabled workloads set the head_gpus and num_gpus\n",
6465
" namespace=namespace,\n",
6566
" num_workers=2,\n",
6667
" min_cpus=2,\n",

0 commit comments

Comments
 (0)