diff --git a/.azure/hpu-tests.yml b/.azure/hpu-tests.yml index 2935f72e2b..d8eef9d608 100644 --- a/.azure/hpu-tests.yml +++ b/.azure/hpu-tests.yml @@ -39,7 +39,7 @@ jobs: cancelTimeoutInMinutes: "2" pool: intel-hpus container: - image: "vault.habana.ai/gaudi-docker/1.7.0/ubuntu20.04/habanalabs/pytorch-installer-1.12.0:latest" + image: "vault.habana.ai/gaudi-docker/1.7.1/ubuntu20.04/habanalabs/pytorch-installer-1.13.0:latest" options: "--runtime=habana -e HABANA_VISIBLE_DEVICES=all -e OMPI_MCA_btl_vader_single_copy_mechanism=none --cap-add=sys_nice --ipc=host --shm-size=4g -v /usr/bin/docker:/tmp/docker:ro" workspace: clean: all diff --git a/dockers/ci-runner-hpu/Dockerfile b/dockers/ci-runner-hpu/Dockerfile index db9f911312..4e9a191efd 100644 --- a/dockers/ci-runner-hpu/Dockerfile +++ b/dockers/ci-runner-hpu/Dockerfile @@ -16,8 +16,8 @@ # gaudi-docker-agent:latest ARG DIST="latest" -ARG GAUDI_VERSION="1.7.0" -ARG PYTORCH_INSTALLER_VERSION="1.12.0" +ARG GAUDI_VERSION="1.7.1" +ARG PYTORCH_INSTALLER_VERSION="1.13.0" FROM vault.habana.ai/gaudi-docker/${GAUDI_VERSION}/ubuntu20.04/habanalabs/pytorch-installer-${PYTORCH_INSTALLER_VERSION}:${DIST} LABEL maintainer="https://vault.habana.ai/" diff --git a/src/pytorch_lightning/utilities/imports.py b/src/pytorch_lightning/utilities/imports.py index f2efdfcb82..1fcbbbe501 100644 --- a/src/pytorch_lightning/utilities/imports.py +++ b/src/pytorch_lightning/utilities/imports.py @@ -52,6 +52,8 @@ if _HABANA_FRAMEWORK_AVAILABLE: from habana_frameworks.torch.utils.library_loader import is_habana_avaialble _HPU_AVAILABLE = is_habana_avaialble() + if _HPU_AVAILABLE: + _TORCH_GREATER_EQUAL_1_13 = compare_version("torch", operator.ge, "1.13.0", use_base_version=True) else: _HPU_AVAILABLE = False