These PyTorch* container images contain the Intel® Gaudi® software and PyTorch* framework for running models on the Intel Gaudi AI accelerator. For more details on the container runtime and installing Docker images, see the installation guide.
Pull Commands
PyTorch 2.6.0, Intel® Gaudi® Software 1.20.0, Ubuntu* 24.04 (Intel Gaudi 2 and Intel Gaudi 3 Accelerators)
docker pull vault.habana.ai/gaudi-docker/1.20.0/ubuntu24.04/habanalabs/pytorch-installer-2.6.0:latest
PyTorch 2.6.0, Intel Gaudi Software 1.20.0, Ubuntu 22.04 (Intel Gaudi 2 and Intel Gaudi 3 Accelerators)
docker pull vault.habana.ai/gaudi-docker/1.20.0/ubuntu22.04/habanalabs/pytorch-installer-2.6.0:latest
PyTorch 2.6.0, Intel Gaudi Software 1.20.0, Red Hat Enterprise Linux(RHEL)* 9.4 (Intel Gaudi 2 and Intel Gaudi 3 Accelerators)
docker pull vault.habana.ai/gaudi-docker/1.20.0/rhel9.4/habanalabs/pytorch-installer-2.6.0:latest
PyTorch 2.6.0, Intel Gaudi Software 1.20.0, RHEL9.2 (Intel Gaudi 2)
docker pull vault.habana.ai/gaudi-docker/1.20.0/rhel9.2/habanalabs/pytorch-installer-2.6.0:latest
PyTorch 2.6.0, Intel Gaudi Software 1.20.0, RHEL8.6 (Intel Gaudi 2)
docker pull vault.habana.ai/gaudi-docker/1.20.0/rhel8.6/habanalabs/pytorch-installer-2.6.0:latest
PyTorch 2.6.0, Intel Gaudi Software 1.20.0, Tencent* 3.1 (Intel Gaudi 2)
docker pull vault.habana.ai/gaudi-docker/1.20.0/tencentOS3.1/habanalabs/pytorch-installer-2.6.0:latest
PyTorch 2.6.0, Intel Gaudi Software 1.20.0, Amazon* for Linux* 2 (First Generation Intel Gaudi Accelerator Only)
docker pull vault.habana.ai/gaudi-docker/1.20.0/amzn2/habanalabs/pytorch-installer-2.6.0:latest