diff --git a/Dockerfile_gaudi b/Dockerfile_gaudi index e6c40d68..bd6c58b4 100644 --- a/Dockerfile_gaudi +++ b/Dockerfile_gaudi @@ -1,5 +1,5 @@ # Those arguments are required to build the image -ARG HABANA_VERSION=1.20.0 +ARG HABANA_VERSION=1.21.0 ARG PYTORCH_VERSION=2.6.0 # Rust builder diff --git a/backends/gaudi/Makefile b/backends/gaudi/Makefile index c153a5ff..77581517 100644 --- a/backends/gaudi/Makefile +++ b/backends/gaudi/Makefile @@ -2,7 +2,7 @@ mkfile_path := $(abspath $(lastword $(MAKEFILE_LIST))) mkfile_dir := $(dir $(mkfile_path)) root_dir := ${mkfile_dir}/../.. -HABANA_VERSION := 1.20.0 +HABANA_VERSION := 1.21.0 PYTORCH_VERSION := 2.6.0 .PHONY: image run-local-dev-container install-dependencies install-server install-router install-launcher local-dev-install