Skip to content

Migrate gsutil usage to gcloud storage #4329

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
gericdong merged 12 commits into from
Dec 25, 2025
Merged

Migrate gsutil usage to gcloud storage #4329

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
d38ac1c
Migrate gsutil usage to gcloud storage
googlyrahman Oct 17, 2025
d8571f8
Manual Changes
bhandarivijay-png Dec 12, 2025
8ca9b56
Manual Changes
bhandarivijay-png Dec 12, 2025
a1ed9c6
Linter fiex the issues for 4329
bhandarivijay-png Dec 17, 2025
a7ab350
Revert "Linter fiex the issues for 4329"
bhandarivijay-png Dec 17, 2025
25c3780
Revert "Manual Changes"
bhandarivijay-png Dec 22, 2025
682761f
Merge pull request #39 from bhandarivijay-png/ai-gsutil-migration-b2f…
gurusai-voleti Dec 22, 2025
41dad18
changes removed model garden
bhandarivijay-png Dec 22, 2025
f2decbe
Merge pull request #96 from bhandarivijay-png/ai-gsutil-migration-b2f…
gurusai-voleti Dec 23, 2025
a42f0a0
Update model_garden_gemma2_finetuning_on_vertex.ipynb
gurusai-voleti Dec 23, 2025
46e3549
Update model_garden_pytorch_llama3_3_finetuning.ipynb
gurusai-voleti Dec 23, 2025
1496941
Update model_garden_pytorch_llama3_3_finetuning.ipynb
gurusai-voleti Dec 23, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 4 additions & 8 deletions notebooks/community/ml_ops/stage2/get_started_with_tabnet.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -467,8 +467,7 @@
},
"outputs": [],
"source": [
"! gsutil mb -l $REGION $BUCKET_URI"
]
"! gcloud storage buckets create --location $REGION $BUCKET_URI" ]
},
{
"cell_type": "markdown",
Expand All @@ -487,8 +486,7 @@
},
"outputs": [],
"source": [
"! gsutil ls -al $BUCKET_URI"
]
"! gcloud storage ls --all-versions --long $BUCKET_URI" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -711,8 +709,7 @@
"IMPORT_FILE = \"petfinder-tabular-classification-tabnet-with-header.csv\"\n",
"TRAINING_DATA_PATH = f\"{BUCKET_URI}/data/petfinder/train.csv\"\n",
"\n",
"! gsutil cp gs://cloud-samples-data/ai-platform-unified/datasets/tabular/{IMPORT_FILE} {TRAINING_DATA_PATH}"
]
"! gcloud storage cp gs://cloud-samples-data/ai-platform-unified/datasets/tabular/{IMPORT_FILE} {TRAINING_DATA_PATH}" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -1422,8 +1419,7 @@
" print(e)\n",
"\n",
"if delete_bucket or os.getenv(\"IS_TESTING\"):\n",
" ! gsutil rm -r $BUCKET_URI"
]
" ! gcloud storage rm --recursive $BUCKET_URI" ]
}
],
"metadata": {
Expand Down
31 changes: 11 additions & 20 deletions notebooks/community/ml_ops/stage3/get_started_with_hpt_pipeline_components.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -441,8 +441,7 @@
},
"outputs": [],
"source": [
"! gsutil mb -l $REGION $BUCKET_URI"
]
"! gcloud storage buckets create $BUCKET_URI --location=$REGION" ]
},
{
"cell_type": "markdown",
Expand All @@ -461,8 +460,7 @@
},
"outputs": [],
"source": [
"! gsutil ls -al $BUCKET_URI"
]
"! gcloud storage ls --all-versions --long $BUCKET_URI" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -531,10 +529,9 @@
},
"outputs": [],
"source": [
"! gsutil iam ch serviceAccount:{SERVICE_ACCOUNT}:roles/storage.objectCreator $BUCKET_URI\n",
"\n",
"! gsutil iam ch serviceAccount:{SERVICE_ACCOUNT}:roles/storage.objectViewer $BUCKET_URI"
]

"! gcloud storage buckets add-iam-policy-binding $BUCKET_URI --member=serviceAccount:{SERVICE_ACCOUNT} --role=roles/storage.objectCreator\n", "\n",
"! gcloud storage buckets add-iam-policy-binding $BUCKET_URI --member=serviceAccount:{SERVICE_ACCOUNT} --role=roles/storage.objectViewer" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -1006,8 +1003,7 @@
"! rm -f custom.tar custom.tar.gz\n",
"! tar cvf custom.tar custom\n",
"! gzip custom.tar\n",
"! gsutil cp custom.tar.gz $BUCKET_URI/trainer_horses_or_humans.tar.gz"
]
"! gcloud storage cp custom.tar.gz $BUCKET_URI/trainer_horses_or_humans.tar.gz" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -1433,14 +1429,11 @@
" + \"/evaluation_metrics\"\n",
" )\n",
" if tf.io.gfile.exists(EXECUTE_OUTPUT):\n",
" ! gsutil cat $EXECUTE_OUTPUT\n",
" return EXECUTE_OUTPUT\n",
" ! gcloud storage cat $EXECUTE_OUTPUT\n", " return EXECUTE_OUTPUT\n",
" elif tf.io.gfile.exists(GCP_RESOURCES):\n",
" ! gsutil cat $GCP_RESOURCES\n",
" return GCP_RESOURCES\n",
" ! gcloud storage cat $GCP_RESOURCES\n", " return GCP_RESOURCES\n",
" elif tf.io.gfile.exists(EVAL_METRICS):\n",
" ! gsutil cat $EVAL_METRICS\n",
" return EVAL_METRICS\n",
" ! gcloud storage cat $EVAL_METRICS\n", " return EVAL_METRICS\n",
"\n",
" return None\n",
"\n",
Expand All @@ -1454,8 +1447,7 @@
"print(\"getbesttrialop\")\n",
"artifacts = print_pipeline_output(pipeline, \"getbesttrialop\")\n",
"print(\"\\n\\n\")\n",
"output = !gsutil cat $artifacts\n",
"output = json.loads(output[0])\n",
"output = !gcloud storage cat $artifacts\n", "output = json.loads(output[0])\n",
"best_trial = json.loads(output[\"parameters\"][\"Output\"][\"stringValue\"])\n",
"model_id = best_trial[\"id\"]\n",
"print(\"BEST MODEL\", model_id)\n",
Expand Down Expand Up @@ -1529,8 +1521,7 @@
"delete_bucket = False\n",
"\n",
"if delete_bucket or os.getenv(\"IS_TESTING\"):\n",
" ! gsutil rm -r $BUCKET_URI"
]
" ! gcloud storage rm --recursive $BUCKET_URI" ]
}
],
"metadata": {
Expand Down
11 changes: 4 additions & 7 deletions notebooks/community/ml_ops/stage5/get_started_with_vertex_endpoint_and_shared_vm.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -449,8 +449,7 @@
},
"outputs": [],
"source": [
"! gsutil mb -l $REGION $BUCKET_URI"
]
"! gcloud storage buckets create --location=$REGION $BUCKET_URI" ]
},
{
"cell_type": "markdown",
Expand All @@ -469,8 +468,7 @@
},
"outputs": [],
"source": [
"! gsutil ls -al $BUCKET_URI"
]
"! gcloud storage ls --all-versions --long $BUCKET_URI" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -1374,7 +1372,7 @@
},
"outputs": [],
"source": [
"! gsutil cp gs://cloud-ml-data/img/flower_photos/daisy/100080576_f52e8ee070_n.jpg test.jpg\n",
"! gcloud storage cp gs://cloud-ml-data/img/flower_photos/daisy/100080576_f52e8ee070_n.jpg test.jpg\n",
"\n",
"import base64\n",
"\n",
Expand Down Expand Up @@ -1574,8 +1572,7 @@
"delete_bucket = False\n",
"\n",
"if delete_bucket or os.getenv(\"IS_TESTING\"):\n",
" ! gsutil rm -r $BUCKET_URI\n",
"\n",
" ! gcloud storage rm --recursive $BUCKET_URI\n", "\n",
"!rm -f test.jpg"
]
}
Expand Down
12 changes: 4 additions & 8 deletions notebooks/community/ml_ops/stage6/get_started_with_custom_image_model_batch.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -455,8 +455,7 @@
},
"outputs": [],
"source": [
"! gsutil mb -l $REGION $BUCKET_URI"
]
"! gcloud storage buckets create --location $REGION $BUCKET_URI" ]
},
{
"cell_type": "markdown",
Expand All @@ -475,8 +474,7 @@
},
"outputs": [],
"source": [
"! gsutil ls -al $BUCKET_URI"
]
"! gcloud storage ls --all-versions --long $BUCKET_URI" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -927,8 +925,7 @@
" f.write(json.dumps(x.tolist()) + \"\\n\")\n",
"\n",
"# Upload to Cloud Storage bucket\n",
"! gsutil cp $BATCH_PREDICTION_INSTANCES_FILE $BATCH_PREDICTION_GCS_SOURCE\n",
"\n",
"! gcloud storage cp $BATCH_PREDICTION_INSTANCES_FILE $BATCH_PREDICTION_GCS_SOURCE\n", "\n",
"print(\"Uploaded instances to: \", BATCH_PREDICTION_GCS_SOURCE)"
]
},
Expand Down Expand Up @@ -1445,8 +1442,7 @@
" batch_prediction_job.delete()\n",
"\n",
"if delete_bucket or os.getenv(\"IS_TESTING\"):\n",
" ! gsutil rm -rf {BUCKET_URI}"
]
" ! gcloud storage rm --recursive --continue-on-error {BUCKET_URI}" ]
}
],
"metadata": {
Expand Down
13 changes: 4 additions & 9 deletions .../community/prediction/custom_prediction_routines/SDK_Custom_Predict_SDK_Integration.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -496,8 +496,7 @@
},
"outputs": [],
"source": [
"! gsutil mb -l $REGION -p $PROJECT_ID $BUCKET_URI"
]
"! gcloud storage buckets create --location=$REGION --project=$PROJECT_ID $BUCKET_URI" ]
},
{
"cell_type": "markdown",
Expand All @@ -516,8 +515,7 @@
},
"outputs": [],
"source": [
"! gsutil ls -al $BUCKET_URI"
]
"! gcloud storage ls --all-versions --long $BUCKET_URI" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -687,9 +685,7 @@
"outputs": [],
"source": [
"%cd ..\n",
"!gsutil cp {LOCAL_MODEL_ARTIFACTS_DIR}/* {BUCKET_URI}/{MODEL_ARTIFACT_DIR}/\n",
"!gsutil ls {BUCKET_URI}/{MODEL_ARTIFACT_DIR}/"
]
"!gcloud storage cp {LOCAL_MODEL_ARTIFACTS_DIR}/* {BUCKET_URI}/{MODEL_ARTIFACT_DIR}/\n", "!gcloud storage ls {BUCKET_URI}/{MODEL_ARTIFACT_DIR}/" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -1792,8 +1788,7 @@
"delete_bucket = False\n",
"\n",
"if delete_bucket or os.getenv(\"IS_TESTING\"):\n",
" ! gsutil rm -r $BUCKET_URI"
]
" ! gcloud storage rm --recursive $BUCKET_URI" ]
}
],
"metadata": {
Expand Down
18 changes: 6 additions & 12 deletions ...community/prediction/feature_store_integration/prediction_feature_store_integration.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -451,8 +451,7 @@
},
"outputs": [],
"source": [
"! gsutil mb -l $REGION -p $PROJECT_ID $BUCKET_URI"
]
"! gcloud storage buckets create --location=$REGION --project=$PROJECT_ID $BUCKET_URI" ]
},
{
"cell_type": "code",
Expand All @@ -462,8 +461,7 @@
},
"outputs": [],
"source": [
"! gsutil ls -al $BUCKET_URI"
]
"! gcloud storage ls --all-versions --long $BUCKET_URI" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -493,8 +491,7 @@
},
"outputs": [],
"source": [
"! gsutil cp -r gs://featurestore_integration/model/* $ARTIFACT_URI"
]
"! gcloud storage cp --recursive gs://featurestore_integration/model/* $ARTIFACT_URI" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -855,8 +852,7 @@
},
"outputs": [],
"source": [
"! gsutil cp gs://featurestore_integration/data/mobile_gaming_dataset.csv $SOURCE_URI"
]
"! gcloud storage cp gs://featurestore_integration/data/mobile_gaming_dataset.csv $SOURCE_URI" ]
},
{
"cell_type": "code",
Expand Down Expand Up @@ -1198,8 +1194,7 @@
},
"outputs": [],
"source": [
"!gsutil cp prediction_featurestore_fetch_config.yaml $ARTIFACT_URI"
]
"!gcloud storage cp prediction_featurestore_fetch_config.yaml $ARTIFACT_URI" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -1731,8 +1726,7 @@
"outputs": [],
"source": [
"# Delete bucket\n",
"!gsutil -m rm -r $BUCKET_URI"
]
"!gcloud storage rm --recursive $BUCKET_URI" ]
}
],
"metadata": {
Expand Down
9 changes: 3 additions & 6 deletions notebooks/official/workbench/spark/spark_ml.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -383,8 +383,7 @@
},
"outputs": [],
"source": [
"! gsutil mb -l {LOCATION} -p {PROJECT_ID} {BUCKET_URI}"
]
"! gcloud storage buckets create --location {LOCATION} --project {PROJECT_ID} {BUCKET_URI}" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -1424,8 +1423,7 @@
" ! rm -rf $BUCKET_FOLDER\n",
" ! mkdir $BUCKET_FOLDER\n",
" bike_gbt_model.write().overwrite().save(f\"{BUCKET_FOLDER}\")\n",
" ! gsutil cp -r $BUCKET_FOLDER $BUCKET_URI\n",
"else:\n",
" ! gcloud storage cp --recursive $BUCKET_FOLDER $BUCKET_URI\n", "else:\n",
" bike_gbt_model.write().overwrite().save(f\"{BUCKET_URI}/\")"
]
},
Expand Down Expand Up @@ -1460,8 +1458,7 @@
"# Delete the Cloud Storage bucket\n",
"delete_bucket = False # Set True for deletion\n",
"if delete_bucket:\n",
" ! gsutil rm -r $BUCKET_URI"
]
" ! gcloud storage rm --recursive $BUCKET_URI" ]
},
{
"cell_type": "markdown",
Expand Down
Loading