From 522a576f88ec2debe9faaa2663ec4322f0b1691e Mon Sep 17 00:00:00 2001 From: Takumi Ohyama Date: Fri, 19 Apr 2024 15:39:32 +0000 Subject: [PATCH 1/2] disable GPU from ponlKFP prebuilt pipeline --- .../pipelines/labs/kfp_pipeline_vertex_prebuilt.ipynb | 5 +++-- .../pipelines/solutions/kfp_pipeline_vertex_prebuilt.ipynb | 5 +++-- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/notebooks/kubeflow_pipelines/pipelines/labs/kfp_pipeline_vertex_prebuilt.ipynb b/notebooks/kubeflow_pipelines/pipelines/labs/kfp_pipeline_vertex_prebuilt.ipynb index 1b11c30f..113354af 100644 --- a/notebooks/kubeflow_pipelines/pipelines/labs/kfp_pipeline_vertex_prebuilt.ipynb +++ b/notebooks/kubeflow_pipelines/pipelines/labs/kfp_pipeline_vertex_prebuilt.ipynb @@ -250,8 +250,9 @@ " {\n", " \"machine_spec\": {\n", " \"machine_type\": \"n1-standard-4\",\n", - " \"accelerator_type\": \"NVIDIA_TESLA_T4\",\n", - " \"accelerator_count\": 1,\n", + " # Enable if you want to use GPU.\n", + " # \"accelerator_type\": \"NVIDIA_TESLA_T4\",\n", + " # \"accelerator_count\": 1,\n", " },\n", " \"replica_count\": 1,\n", " \"container_spec\": {\n", diff --git a/notebooks/kubeflow_pipelines/pipelines/solutions/kfp_pipeline_vertex_prebuilt.ipynb b/notebooks/kubeflow_pipelines/pipelines/solutions/kfp_pipeline_vertex_prebuilt.ipynb index c9453ae5..a1c2e970 100644 --- a/notebooks/kubeflow_pipelines/pipelines/solutions/kfp_pipeline_vertex_prebuilt.ipynb +++ b/notebooks/kubeflow_pipelines/pipelines/solutions/kfp_pipeline_vertex_prebuilt.ipynb @@ -240,8 +240,9 @@ " {\n", " \"machine_spec\": {\n", " \"machine_type\": \"n1-standard-4\",\n", - " \"accelerator_type\": \"NVIDIA_TESLA_T4\",\n", - " \"accelerator_count\": 1,\n", + " # Enable if you want to use GPU.\n", + " # \"accelerator_type\": \"NVIDIA_TESLA_T4\",\n", + " # \"accelerator_count\": 1,\n", " },\n", " \"replica_count\": 1,\n", " \"container_spec\": {\n", From ed792fd6088ecae6551f1d0706224dea6f6531ce Mon Sep 17 00:00:00 2001 From: Takumi Ohyama Date: Fri, 19 Apr 2024 15:46:33 +0000 Subject: [PATCH 2/2] disabled GPU on KFP CICD --- .../labs/pipeline_vertex/tuning_lightweight_component.py | 5 +++-- .../pipeline_vertex/tuning_lightweight_component.py | 5 +++-- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/notebooks/kubeflow_pipelines/cicd/labs/pipeline_vertex/tuning_lightweight_component.py b/notebooks/kubeflow_pipelines/cicd/labs/pipeline_vertex/tuning_lightweight_component.py index 98c23093..1c4764c4 100644 --- a/notebooks/kubeflow_pipelines/cicd/labs/pipeline_vertex/tuning_lightweight_component.py +++ b/notebooks/kubeflow_pipelines/cicd/labs/pipeline_vertex/tuning_lightweight_component.py @@ -46,8 +46,9 @@ def tune_hyperparameters( { "machine_spec": { "machine_type": "n1-standard-4", - "accelerator_type": "NVIDIA_TESLA_K80", - "accelerator_count": 1, + # Enable if you want to use GPU. + # "accelerator_type": "NVIDIA_TESLA_K80", + # "accelerator_count": 1, }, "replica_count": 1, "container_spec": { diff --git a/notebooks/kubeflow_pipelines/cicd/solutions/pipeline_vertex/tuning_lightweight_component.py b/notebooks/kubeflow_pipelines/cicd/solutions/pipeline_vertex/tuning_lightweight_component.py index 98c23093..1c4764c4 100644 --- a/notebooks/kubeflow_pipelines/cicd/solutions/pipeline_vertex/tuning_lightweight_component.py +++ b/notebooks/kubeflow_pipelines/cicd/solutions/pipeline_vertex/tuning_lightweight_component.py @@ -46,8 +46,9 @@ def tune_hyperparameters( { "machine_spec": { "machine_type": "n1-standard-4", - "accelerator_type": "NVIDIA_TESLA_K80", - "accelerator_count": 1, + # Enable if you want to use GPU. + # "accelerator_type": "NVIDIA_TESLA_K80", + # "accelerator_count": 1, }, "replica_count": 1, "container_spec": {