Self-managed installation and maintenance > Installation and configuration guide > Google Kubernetes Engine (GKE) > DataRobot container images

DataRobot container images¶

The DataRobot platform uses Docker Hub as its primary container registry. While the DataRobot Helm chart is configured to use Docker Hub by default, it also supports using a dedicated private GAR container registry if required.

Note

For additional information about Docker Hub, see the Docker Hub documentation.

Using private Google Artifact Registry (GAR)¶

You can install the DataRobot platform by pulling application images from either Docker Hub or a private Google Artifact Registry (GAR). To use a private GAR Registry to store the DataRobot platform images, follow the steps listed below.

Set environment variables¶

To store configuration values, set the following variables (ensure that the below commands include the necessary context):

export DOCKERHUB_USERNAME="DATAROBOT_CUSTOMER_USERNAME"
export DOCKERHUB_PASSWORD="DATAROBOT_CUSTOMER_PASSWORD"
export DATAROBOT_VERSION="X.X.X"
export PROJECT_NAME="GCP_PROJECT_NAME"
export REGION="GCP_REGION"
export REPO="GCP_REPO_NAME"
export BASE64_SERVICE_ACCOUNT_KEY="GCP_BASE64_SERVICE_ACCOUNT_KEY"

Note

Replace DATAROBOT_CUSTOMER_USERNAME and DATAROBOT_CUSTOMER_PASSWORD with your Docker Hub credentials. You can request these credentials through the DataRobot Support Portal or by emailing support@datarobot.com.
Replace X.X.X with the latest release chart version.
Replace GCP_PROJECT_NAME with your actual Google Cloud Project name.
Reaplce GCP_REGION with your actual Google region used to deploy the resources.
Reaplce GCP_REPO_NAME with your actual GAR repository name.
Reaplce GCP_BASE64_SERVICE_ACCOUNT_KEY with your actual Service Key encoded as base64.

Initial setup¶

To authenticate your Docker Hub session, run the following command:

echo ${DOCKERHUB_PASSWORD} | docker login -u ${DOCKERHUB_USERNAME} --password-stdin

Download the main DataRobot chart from Docker Hub:

echo ${DOCKERHUB_PASSWORD} | helm registry login registry-1.docker.io -u ${DOCKERHUB_USERNAME} --password-stdin
helm pull oci://registry-1.docker.io/datarobot/datarobot-prime --version ${DATAROBOT_VERSION}

The process of loading images into your private registry is handled by a dedicated Helm plugin. To install the plugin, use the following command:

helm plugin install https://github.com/datarobot-oss/helm-datarobot-plugin.git

Load images into GAR¶

Load images into your private registry by pulling DataRobot images from Docker Hub:

export PREFIX="${PROJECT_NAME}/${REPO}"
helm datarobot sync datarobot-prime-${DATAROBOT_VERSION}.tgz -r "${REGION}-docker.pkg.dev" --prefix $PREFIX -u _json_key_base64 -p "${BASE64_SERVICE_ACCOUNT_KEY}"

If it is a restricted network, you can load images using the compressed tarball artifact:

export PREFIX="${PROJECT_NAME}/${REPO}"
helm datarobot load tarball-datarobot-${DATAROBOT_VERSION}.tar.zst -r "${REGION}-docker.pkg.dev" --prefix $PREFIX -u _json_key_base64 -p "${BASE64_SERVICE_ACCOUNT_KEY}"

Verify images are in GAR¶

To confirm that the repositories have been created and the images have loaded into Google Artifact Registry, the following command, replacing the example path with your own:

gcloud artifacts docker images list ${REGION}.pkg.dev/${PROJECT_NAME}/${REPO}/

The output will list the images in the specified repository.

Listing items under project example-google-project, location us-east1, repository example-datarobot-registry.

IMAGE                                                                                             DIGEST                                                                   CREATE_TIME          UPDATE_TIME
us-east1-docker.pkg.dev/example-google-project/example-datarobot-registry/datarobot-runtime       sha256:482aabdceb859cc9ddc2e36867ac1ff6d3477bffcd05cac015bfc442c334aff4  2024-07-23T17:34:00  2024-07-23T17:34:00
...