Upgrade to HPU release 1.7.1 (#15956)
* 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>
This commit is contained in:
parent
df678335d6
commit
8475f85e16
|
@ -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
|
||||
|
|
|
@ -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/"
|
||||
|
|
|
@ -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
|
||||
|
||||
|
|
Loading…
Reference in New Issue