onnxruntime/tools/ci_build/github/linux/docker/scripts/training
Suffian Khan 008c5f7640
Use single builder image across Python versions for ROCm wheels (#8302)
* first attempt share docker image across python and torch versons

* set dependency between jobs

* fix yaml grammer

* remove python version from first stage

* clean deepspeed directroy

* split into two images according torch version

* fix yaml syntax

* invalidate cache

* remove DS to prevent torch 1.9.0 upgrade
2021-07-06 11:56:00 -07:00
..
ortmodule Use single builder image across Python versions for ROCm wheels (#8302) 2021-07-06 11:56:00 -07:00
azure_scale_set_vm_mount_test_data.sh Cache datasets on CI machines (#6525) 2021-02-02 21:11:35 -08:00
requirements.txt Add post-install command to build PyTorch CPP extensions from within onnxruntime package (#8027) 2021-06-28 18:11:58 -07:00