mirror of
https://github.com/deepspeedai/DeepSpeed.git
synced 2025-10-20 15:33:51 +08:00
Update gaudi2 nightly,ci to latest 1.21.0 build (#7313)
Signed-off-by: Shaik Raza Sikander <srsikander@habana.ai> Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
This commit is contained in:
2
.github/workflows/hpu-gaudi2-nightly.yml
vendored
2
.github/workflows/hpu-gaudi2-nightly.yml
vendored
@ -21,7 +21,7 @@ jobs:
|
|||||||
# The type of runner that the job will run on
|
# The type of runner that the job will run on
|
||||||
runs-on: [self-hosted, intel, gaudi2]
|
runs-on: [self-hosted, intel, gaudi2]
|
||||||
container:
|
container:
|
||||||
image: vault.habana.ai/gaudi-docker/1.20.0/ubuntu22.04/habanalabs/pytorch-installer-2.6.0:latest
|
image: vault.habana.ai/gaudi-docker/1.21.0/ubuntu22.04/habanalabs/pytorch-installer-2.6.0:latest
|
||||||
ports:
|
ports:
|
||||||
- 80
|
- 80
|
||||||
options: --runtime=habana -e HABANA_VISIBLE_DEVICES=all -e OMPI_MCA_btl_vader_single_copy_mechanism=none --cap-add=sys_nice
|
options: --runtime=habana -e HABANA_VISIBLE_DEVICES=all -e OMPI_MCA_btl_vader_single_copy_mechanism=none --cap-add=sys_nice
|
||||||
|
2
.github/workflows/hpu-gaudi2.yml
vendored
2
.github/workflows/hpu-gaudi2.yml
vendored
@ -39,7 +39,7 @@ jobs:
|
|||||||
# The type of runner that the job will run on
|
# The type of runner that the job will run on
|
||||||
runs-on: [self-hosted, intel, gaudi2]
|
runs-on: [self-hosted, intel, gaudi2]
|
||||||
container:
|
container:
|
||||||
image: vault.habana.ai/gaudi-docker/1.20.0/ubuntu22.04/habanalabs/pytorch-installer-2.6.0:latest
|
image: vault.habana.ai/gaudi-docker/1.21.0/ubuntu22.04/habanalabs/pytorch-installer-2.6.0:latest
|
||||||
ports:
|
ports:
|
||||||
- 80
|
- 80
|
||||||
options: --runtime=habana -e HABANA_VISIBLE_DEVICES=all -e OMPI_MCA_btl_vader_single_copy_mechanism=none --cap-add=sys_nice
|
options: --runtime=habana -e HABANA_VISIBLE_DEVICES=all -e OMPI_MCA_btl_vader_single_copy_mechanism=none --cap-add=sys_nice
|
||||||
|
Reference in New Issue
Block a user