Skip to content

Commit

Permalink
Upgrade to HPU release 1.7.1 (#15956)
Browse files Browse the repository at this point in the history
* Upgrade to HPU release 1.7.1
Update torch version check for hpu

Signed-off-by: Jerome <janand@habana.ai>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
  • Loading branch information
jerome-habana and pre-commit-ci[bot] committed Dec 8, 2022
1 parent df67833 commit 8475f85
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 3 deletions.
2 changes: 1 addition & 1 deletion .azure/hpu-tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
4 changes: 2 additions & 2 deletions dockers/ci-runner-hpu/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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/"
Expand Down
2 changes: 2 additions & 0 deletions src/pytorch_lightning/utilities/imports.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,8 @@
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

Expand Down

0 comments on commit 8475f85

Please sign in to comment.