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 installation guide on instructions to setup Container Runtime and Installation of docker images.