Fix onnxruntime-CI-nightly-ort-pipeline Failure (#16495)

The image for the onnxruntime-CI-nightly-ort-pipeline is too old. 
The ort package in the image is older than latest test code in nightly
ci. This causes the nightly ci failed.
This commit is contained in:
guyang3532 2023-06-27 23:19:23 +08:00 committed by GitHub
parent a49bb85cfe
commit 4768ac5f30
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-cu117-py38-torch210dev \
ptebic.azurecr.io/internal/azureml/aifx/nightly-ubuntu2004-cu118-py38-torch210dev \
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()