Merge pull request #447 from GoogleCloudPlatform/disable_gpu_prebuild

Disable GPU on the KFP prebuilt pipeline
GoogleCloudPlatform · May 3, 2024 · cd9ec69 · cd9ec69
2 parents 43c5963 + ed792fd
commit cd9ec69
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 8 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
@@ -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
@@ -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/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
@@ -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",