mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-30 20:18:08 +00:00
nightly pipeline build using PTCA image. (#12605)
* nightly pipeline yaml and requirements files * changed names, removed torchvision installing * delete old file Co-authored-by: Adam Louly <adamlouly@microsoft.com@orttrainingdev7.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net>
This commit is contained in:
parent
0757d51334
commit
94f76b944e
2 changed files with 31 additions and 0 deletions
|
|
@ -0,0 +1,28 @@
|
|||
trigger: none
|
||||
|
||||
jobs:
|
||||
- job: Onnxruntime_Linux_Nightly_ORTModule_tests
|
||||
|
||||
timeoutInMinutes: 120
|
||||
pool: 'Onnxruntime-Linux-GPU-NC6sv3'
|
||||
|
||||
steps:
|
||||
- checkout: self
|
||||
clean: true
|
||||
submodules: recursive
|
||||
|
||||
# Entry point for all ortmodule training tests
|
||||
- script: |
|
||||
docker run \
|
||||
--gpus all \
|
||||
--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-cu113-py38-torch1130dev \
|
||||
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()
|
||||
timeoutInMinutes: 120
|
||||
- template: templates/component-governance-component-detection-steps.yml
|
||||
parameters:
|
||||
condition: 'succeeded'
|
||||
|
|
@ -0,0 +1,3 @@
|
|||
sklearn
|
||||
transformers==v4.4.2
|
||||
wget
|
||||
Loading…
Reference in a new issue