Integrations
Google Artifact Registry
and
ZenML logo in purple, representing machine learning pipelines and MLOps framework.
Leverage Google’s Container Registry in ZenML Pipelines with Google Artifact Registry Integration
The image is blank. No elements are visible for description or keyword inclusion.
Google Artifact Registry
All integrations

Google Artifact Registry

Leverage Google’s Container Registry in ZenML Pipelines with Google Artifact Registry Integration
Add to ZenML

Leverage Google’s Container Registry in ZenML Pipelines with Google Artifact Registry Integration

Seamlessly integrate the Google Artifact Registry with ZenML. Store, manage, and deploy containerized components within your ML pipelines, leveraging the scalability and reliability of Google Cloud Platform without exposing the complexity to your data scientists.

Features with ZenML

  • Seamless integration of Google Artifact Registry as a container registry in ZenML stacks
  • Effortless storage and retrieval of containerized pipeline environments
  • Scalable and reliable container management backed by Google Cloud infrastructure
  • Streamlined deployment of ML pipelines with versioned container images

Main Features

  • Centralized container image storage and management
  • No local credentials necessary
  • Secure access control and permissions management
  • Automatic versioning and rebuilding of step docker images where necessary

How to use ZenML with
Google Artifact Registry

# Step 1: Install all required gcp packaes 
zenml integration install gcp

# Step 2: Register the GCP container registry
zenml container-registry register gcp_registry \
  --flavor=gcp \
  --uri="<YOUR URI>"

# Step 3: Update your stack to use the selected container registry
# Make sure your stack already contains a remote artifact store
# and orchestrator
zenml stack update -c gcp_registry

# Step 4: Set up authentication (choose one method)
zenml container-registry connect gcp_registry -i

# Step 5: Validate that your stack has a remote orchestrator in it
# Not all orchestrators require a built image, so in order to use the
# container registry you would need a remote orchestrator/step operator
# used in your stack
zenml stack describe

from zenml import pipeline, step

@step
def example_step():
    print("This step will be containerized and pushed to GCP container registry")

@pipeline
def my_pipeline():
    example_step()

if __name__ == "__main__":
    my_pipeline()
    

This code example demonstrates how to integrate Google Artifact Registry with ZenML:

  1. Install the GCP integration
  2. Register the Google Artifact Registry with the URI
  3. Update the ZenML stack to use the registered container registry
  4. Set up authentication using one of the available methods
  5. Define a step and pipeline that will use the Google Artifact Registry for containerized components

When the pipeline is run, the step will be containerized and the image will be pushed to the specified Google Artifact Registry, enabling seamless integration with the GCP ecosystem.

Additional Resources
Google Artifact Registry Documentation
Using Artifact Registry with Google Cloud

Leverage Google’s Container Registry in ZenML Pipelines with Google Artifact Registry Integration

Seamlessly integrate the Google Artifact Registry with ZenML. Store, manage, and deploy containerized components within your ML pipelines, leveraging the scalability and reliability of Google Cloud Platform without exposing the complexity to your data scientists.
Google Artifact Registry

Start Your Free Trial Now

No new paradigms - Bring your own tools and infrastructure
No data leaves your servers, we only track metadata
Free trial included - no strings attached, cancel anytime
Dashboard displaying machine learning models, including versions, authors, and tags. Relevant to model monitoring and ML pipelines.

Connect Your ML Pipelines to a World of Tools

Expand your ML pipelines with Apache Airflow and other 50+ ZenML Integrations
Google Cloud
Skypilot VM
Elastic Container Registry
NeuralProphet
Kaniko
Sagemaker Pipelines
Feast
Modal
XGBoost
Lightning AI
GitHub Container Registry