Home » Catalog » Containers » PyTorch

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.

Pull Command

PyTorch 2.0.1, SynapseAI 1.11.0, Amazon Linux 2
docker pull vault.habana.ai/gaudi-docker/1.11.0/amzn2/habanalabs/pytorch-installer-2.0.1:latest
PyTorch 2.0.1, SynapseAI 1.11.0, Ubuntu 22.04
docker pull vault.habana.ai/gaudi-docker/1.11.0/ubuntu22.04/habanalabs/pytorch-installer-2.0.1:latest
PyTorch 2.0.1, SynapseAI 1.11.0, Ubuntu 20.04
docker pull vault.habana.ai/gaudi-docker/1.11.0/ubuntu20.04/habanalabs/pytorch-installer-2.0.1:latest
PyTorch 2.0.1, SynapseAI 1.11.0, RHEL8.6
docker pull vault.habana.ai/gaudi-docker/1.11.0/rhel8.6/habanalabs/pytorch-installer-2.0.1:latest

Overview

Sign up for the latest Habana developer news, events, training, and updates.