зеркало из https://github.com/microsoft/DeepSpeed.git
Update Gaudi2 docker image (#6677)
This commit is contained in:
Родитель
229960a5e9
Коммит
b3e959460b
|
@ -39,7 +39,7 @@ jobs:
|
|||
# The type of runner that the job will run on
|
||||
runs-on: [self-hosted, intel, gaudi2]
|
||||
container:
|
||||
image: vault.habana.ai/gaudi-docker/1.17.0/ubuntu22.04/habanalabs/pytorch-installer-2.3.1:latest
|
||||
image: vault.habana.ai/gaudi-docker/1.17.1/ubuntu22.04/habanalabs/pytorch-installer-2.3.1:latest
|
||||
ports:
|
||||
- 80
|
||||
options: --runtime=habana -e HABANA_VISIBLE_DEVICES=all -e OMPI_MCA_btl_vader_single_copy_mechanism=none --cap-add=sys_nice
|
||||
|
|
Загрузка…
Ссылка в новой задаче