0.23.0
Search…
⌃K
Links

Installation

How to install ZenML
ZenML is a Python package that can be installed directly via pip:
pip install zenml
Please note that ZenML currently only supports Python 3.7, 3.8, and 3.9. Please adjust your Python environment accordingly.
ZenML comes bundled with a React-based dashboard that lives inside a sister repository. In order to get access to the dashboard locally, you need to launch the ZenML Server and Dashboard locally. For this, you need to install the optional dependencies for the ZenML Server:
pip install "zenml[server]"
If you do not have deployed infrastructure, and want to quickly spin up combinations of tools on the cloud, the MLOps stack sister repository contains a series of Terraform-based recipes to provision such stacks. These recipes can be used directly with ZenML:
pip install "zenml[stacks]"

Virtual Environments

We highly encourage you to install ZenML in a virtual environment. At ZenML, We like to use virtualenvwrapper or pyenv-virtualenv to manage our Python virtual environments.
As mentioned above, make sure that your virtual environment uses one of the supported Python versions.

Verifying Installations

Once the installation is completed, you can check whether the installation was successful through:

Bash

zenml version

Python

import zenml
print(zenml.__version__)
If you would like to learn more about the current release, please visit our PyPi package page.

Known installation issues for M1 Mac Users

Since some of ZenML's dependencies are not compatible with the ARM64 architecture, you need to install ZenML under Rosetta on M1 Macs. See the Installation on M1 Macs page for detailed instructions.

Running with Docker

Python package

zenml is also available as a Docker image hosted publicly on DockerHub. Use the following command to get started in a bash environment with zenml available:
docker run -it zenmldocker/zenml /bin/bash

ZenML Server

If you would like to run the ZenML server with Docker:
docker run -it -d -p 8080:80 zenmldocker/zenml-server

Installing Develop

If you want to use the bleeding edge of ZenML that has not even been released yet, you can install our develop branch directly.
Installing develop is mainly useful if there are key features or bug fixes that you urgently need so you can get those immediately and do not have to wait for the next release.
As the name suggests, the new features in the develop branch are still under development and might not be as polished as the final released version.
Use at your own risk; no guarantees given!
pip install git+https://github.com/zenml-io/[email protected] --upgrade

Using develop with Remote Orchestrators

Remote orchestrators like Kubeflow require Docker Images to set up the environments of each step. By default, they use the official ZenML docker image that we provide with each release. However, if you install from develop, this image will be outdated, so you need to build a custom image instead, and specify it in the configuration of your orchestrator accordingly (see the MLOps Stacks Orchestrator page of your specific orchestrator flavor for more details on how this can be done).

Building Custom Docker Images

To build a custom image, you first need to install Docker, then run one of the following commands from the ZenML repo root depending on your operating system:
Linux, MacOS (Intel), Windows
docker build -t <IMAGE_NAME> -f docker/local-dev.Dockerfile .
MacOS (M1)
docker build --platform linux/amd64 -t <IMAGE_NAME> -f docker/local-dev.Dockerfile .