Skip to main content

DAI Versions

DAI Versions are used to specify the version of Driverless AI to use for DAI engines. The end-user must select a version when creating a DAI engine.

Advanced users and app developers

See the module documentation for full client reference.

Applying DAI Versions​

You can apply a target state of all DAI Versions via configuration. No need to manually Create, Update or Delete all DAI Versions. All existing DAI Versions not specified in the input configuration will be deleted!

import h2o_engine_manager

from h2o_engine_manager.clients.base.image_pull_policy import ImagePullPolicy
from h2o_engine_manager.clients.internal_dai_version.version_config import InternalDAIVersionConfig

# Initialize the client for Driverless AI versions.
client = h2o_engine_manager.login().internal_dai_version_client

configs = [
    InternalDAIVersionConfig(
        internal_dai_version_id="1.10.7",
        image="gcr.io/vorvan/h2oai/dai-x86_64:1.10.7-cuda11.2.2",
        aliases=["latest"],
    ),
    InternalDAIVersionConfig(
        internal_dai_version_id="1.10.6.1",
        image="gcr.io/vorvan/h2oai/dai-x86_64:1.10.6.1-cuda11.2.2",
    ),
    InternalDAIVersionConfig(
        internal_dai_version_id="1.10.6",
        image="gcr.io/vorvan/h2oai/dai-x86_64:1.10.6-cuda11.2.2",
        deprecated=True,
        image_pull_policy=ImagePullPolicy.IMAGE_PULL_POLICY_ALWAYS,
        image_pull_secrets=["secret1", "secret2"],
    )
]

# All other versions not specified in the configs will be deleted.
client.apply_internal_dai_versions(version_configs=configs)
Feedback