Update acpt image in the training pipeline (#14855)

### Description
Current pipeline refers to an old image which is causing test failures.
Updating the image to the latest one.



### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
Fixes pipeline failure:
https://dev.azure.com/onnxruntime/onnxruntime/_build?definitionId=198
- If it fixes an open issue, please link to the issue here. -->
This commit is contained in:
Ashwini Khade 2023-03-07 14:10:32 -08:00 committed by GitHub
parent 5d8ce817cb
commit f71ac9859e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -18,7 +18,7 @@ jobs:
--rm \
--volume $(Build.SourcesDirectory)/orttraining/orttraining/test/python:/onnxruntime_src \
--volume $(Build.SourcesDirectory)/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_nightly:/requirements_torch_nightly \
ptebic.azurecr.io/internal/azureml/aifx/nightly-ubuntu2004-cu116-py38-torch200dev \
ptebic.azurecr.io/internal/azureml/aifx/nightly-ubuntu2004-cu117-py38-torch200dev \
bash -c "python3 -m pip install -r /requirements_torch_nightly/requirements.txt && python3 -m pytest -sv /onnxruntime_src/orttraining_test_ortmodule_api.py"
displayName: 'Run ORTModule Tests'
condition: succeededOrFailed()