Home » Catalog » Containers

Containers

PyTorch

These are the available PyTorch Container images in the Habana Vault. These images contain SynapseAI SW and PyTorch framework that allow a user to run models on Gaudi. For more details on docker usage, please refer to our Setup and Install guide.

TensorFlow

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 Setup and Install guide.
Sign up for the latest Habana developer news, events, training, and updates.