File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
.azure-pipelines/template Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 74
74
75
75
- ${{ if eq(parameters.imageSource, 'pull') }} :
76
76
- script : |
77
- docker pull vault.habana.ai/gaudi-docker/1.16.1/{$OS} /habanalabs/pytorch-installer-2.2.2:latest
77
+ docker pull vault.habana.ai/gaudi-docker/1.16.1/ubuntu22.04 /habanalabs/pytorch-installer-2.2.2:latest
78
78
displayName: "Pull habana docker image"
79
79
80
80
- script : |
95
95
else
96
96
docker run -dit --disable-content-trust --privileged --name=${{ parameters.containerName }} --shm-size="2g" \
97
97
--runtime=habana -e HABANA_VISIBLE_DEVICES=all -e OMPI_MCA_btl_vader_single_copy_mechanism=none --cap-add=sys_nice --net=host --ipc=host \
98
- -v ${BUILD_SOURCESDIRECTORY}:/neural-compressor vault.habana.ai/gaudi-docker/1.16.1/{$OS} /habanalabs/pytorch-installer-2.2.2:latest
98
+ -v ${BUILD_SOURCESDIRECTORY}:/neural-compressor vault.habana.ai/gaudi-docker/1.16.1/ubuntu22.04 /habanalabs/pytorch-installer-2.2.2:latest
99
99
fi
100
100
echo "Show the container list after docker run ... "
101
101
docker ps -a
You can’t perform that action at this time.
0 commit comments