These are the available TensorFlow Container images in the Habana Vault.
These images contain SynapseAI SW and TensorFlow framework that allow a user to run models on Gaudi.
For more details on docker usage, please refer to our Installation Guide on instructions to setup Container Runtime and Installation of docker images
Pull Command
TensorFlow 2.11.0, SynapseAI 1.9.0, Amazon Linux 2
docker pull vault.habana.ai/gaudi-docker/1.9.0/amzn2/habanalabs/tensorflow-installer-tf-cpu-2.11.0:latest
TensorFlow 2.11.0, SynapseAI 1.9.0, Ubuntu 22.04
docker pull vault.habana.ai/gaudi-docker/1.9.0/ubuntu22.04/habanalabs/tensorflow-installer-tf-cpu-2.11.0:latest
TensorFlow 2.11.0, SynapseAI 1.9.0, Ubuntu 20.04
docker pull vault.habana.ai/gaudi-docker/1.9.0/ubuntu20.04/habanalabs/tensorflow-installer-tf-cpu-2.11.0:latest
TensorFlow 2.11.0, SynapseAI 1.9.0, RHEL8.6
docker pull vault.habana.ai/gaudi-docker/1.9.0/rhel8.6/habanalabs/tensorflow-installer-tf-cpu-2.11.0:latest