@@ -80,7 +80,7 @@ Overview
80
80
~~~~~~~~
81
81
This section provides a brief overview of the Vertex SDK for Python. You can also reference the notebooks in `vertex-ai-samples `_ for examples.
82
82
83
- .. _vertex-ai-samples : https://github.com/GoogleCloudPlatform/ai-platform- samples/tree/master/ai-platform-unified/ notebooks/unofficial /sdk
83
+ .. _vertex-ai-samples : https://github.com/GoogleCloudPlatform/vertex- ai-samples/tree/main/ notebooks/community /sdk
84
84
85
85
Importing
86
86
^^^^^^^^^
@@ -201,9 +201,9 @@ It must write the model artifact to the environment variable populated by the tr
201
201
job = aiplatform.CustomTrainingJob(
202
202
display_name = " my-training-job" ,
203
203
script_path = " training_script.py" ,
204
- container_uri = " gcr.io/cloud-aiplatform /training/tf-cpu.2-2:latest" ,
204
+ container_uri = " us-docker.pkg.dev/vertex-ai /training/tf-cpu.2-2:latest" ,
205
205
requirements = [" gcsfs==0.7.1" ],
206
- model_serving_container_image_uri = " gcr.io/cloud-aiplatform /prediction/tf2-cpu.2-2:latest" ,
206
+ model_serving_container_image_uri = " us-docker.pkg.dev/vertex-ai /prediction/tf2-cpu.2-2:latest" ,
207
207
)
208
208
209
209
model = job.run(my_dataset,
@@ -266,7 +266,7 @@ To upload a model:
266
266
model = aiplatform.Model.upload(
267
267
display_name = ' my-model' ,
268
268
artifact_uri = " gs://python/to/my/model/dir" ,
269
- serving_container_image_uri = " gcr.io/cloud-aiplatform /prediction/tf2-cpu.2-2:latest" ,
269
+ serving_container_image_uri = " us-docker.pkg.dev/vertex-ai /prediction/tf2-cpu.2-2:latest" ,
270
270
)
271
271
272
272
To get a model:
0 commit comments