mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-27 20:02:15 +00:00
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:
parent
5d8ce817cb
commit
f71ac9859e
1 changed files with 1 additions and 1 deletions
|
|
@ -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()
|
||||
|
|
|
|||
Loading…
Reference in a new issue