Home » Catalog » Containers » PyTorch Docker Images for the Intel® Gaudi® Accelerator

PyTorch Docker Images for the Intel® Gaudi® Accelerator

These are the available PyTorch Container images in the Intel Gaudi Vault.
These images contain the Intel Gaudi SW and PyTorch framework that allow a user to run models on the Intel Gaudi AI Accelerator
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.3.1, Intel Gaudi Software 1.17.0, Ubuntu 22.04 (Intel Gaudi 2 and Intel Gaudi 3)
docker pull vault.habana.ai/gaudi-docker/1.17.0/ubuntu22.04/habanalabs/pytorch-installer-2.3.1:latest
PyTorch 2.3.1, Intel Gaudi Software 1.17.0, RHEL9.4 (Intel Gaudi 2 and Intel Gaudi 3)
docker pull vault.habana.ai/gaudi-docker/1.17.0/rhel9.4/habanalabs/pytorch-installer-2.3.1:latest
PyTorch 2.3.1, Intel Gaudi Software 1.17.0, RHEL9.2 (Intel Gaudi 2)
docker pull vault.habana.ai/gaudi-docker/1.17.0/rhel9.2/habanalabs/pytorch-installer-2.3.1:latest
PyTorch 2.3.1, Intel Gaudi Software 1.17.0, RHEL8.6 (Intel Gaudi 2)
docker pull vault.habana.ai/gaudi-docker/1.17.0/rhel8.6/habanalabs/pytorch-installer-2.3.1:latest
PyTorch 2.3.1, Intel Gaudi Software 1.17.0, TencentOS 3.1 (Intel Gaudi 2)
docker pull vault.habana.ai/gaudi-docker/1.17.0/tencentOS3.1/habanalabs/pytorch-installer-2.3.1:latest
PyTorch 2.3.1, Intel Gaudi Software 1.17.0, Amazon Linux 2 (first-gen Intel Gaudi only)
docker pull vault.habana.ai/gaudi-docker/1.17.0/amzn2/habanalabs/pytorch-installer-2.3.1:latest

Overview

Stay Informed: Register for the latest Intel Gaudi AI Accelerator developer news, events, training, and updates.