Skip to content

Commit a9de78f

Browse files
raoakarshalexierule
authored andcommitted
CI: HPU support v1.6.0 release (#14794)
* Update hpu-tests.yml to support v1.6.0 release * Update Dockerfile
1 parent f5a1cc4 commit a9de78f

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

.azure/hpu-tests.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ jobs:
3838
cancelTimeoutInMinutes: "2"
3939
pool: habana-gaudi-hpus
4040
container:
41-
image: "vault.habana.ai/gaudi-docker/1.5.0/ubuntu20.04/habanalabs/pytorch-installer-1.11.0:latest"
41+
image: "vault.habana.ai/gaudi-docker/1.6.0/ubuntu20.04/habanalabs/pytorch-installer-1.12.0:latest"
4242
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 --name cd-container -v /usr/bin/docker:/tmp/docker:ro"
4343
workspace:
4444
clean: all

dockers/ci-runner-hpu/Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@
1616
# gaudi-docker-agent:latest
1717

1818
ARG DIST="latest"
19-
ARG GAUDI_VERSION="1.5.0"
20-
ARG PYTORCH_INSTALLER_VERSION="1.11.0"
19+
ARG GAUDI_VERSION="1.6.0"
20+
ARG PYTORCH_INSTALLER_VERSION="1.12.0"
2121
FROM vault.habana.ai/gaudi-docker/${GAUDI_VERSION}/ubuntu20.04/habanalabs/pytorch-installer-${PYTORCH_INSTALLER_VERSION}:${DIST}
2222

2323
LABEL maintainer="https://vault.habana.ai/"

0 commit comments

Comments
 (0)