onnxruntime/tools/ci_build/github/linux/docker/scripts
Suffian Khan 7a3c1787af
Add CI pipeline to publish Python training package targeting Rocm (#7417)
* first attempt rocm training wheel

* modifications needed to python packaging pipeline for Rocm 4.1

* changges to not conflict with cuda

missed stage1 changes

remove package push

add option r to getopt

try again without python install

try again without python install

try again without python install

split pipelines and add back push to remote storage

try on cuda gpu pool

try again

try again

try running without az subscription set

try again on original pipeline

change pool

passing AMD Rocm whl on AMD-GPU pool

split rocm pipeline from cuda pipeline

remove comments

* try adding Rocm tests as well

* try with tests in place

* fix trailing ws

* add training data

* try again as root for tests

* use python3

* typo

* try to map video, render group into container

* try again

* try again

* try to avoid yum error code

* make UID 1001

* try without yum downgrade

* define rocm_version=None

* remove CUDA related comments for Rocm Dockerfile

* Dont pin nightly torch torchvision torchtext versions as they expire (for now nightly is required for Rocm 4.1)

* missed requirements-rocm.txt from last commit

* fix whitespace
2021-04-23 17:22:31 -07:00
..
manylinux pick onnx release candidate (#7177) 2021-04-22 23:57:09 -07:00
training Add CI pipeline to publish Python training package targeting Rocm (#7417) 2021-04-23 17:22:31 -07:00
install_centos.sh
install_deps.sh Add CI pipeline to publish Python training package targeting Rocm (#7417) 2021-04-23 17:22:31 -07:00
install_deps_android.sh
install_openmpi.sh Synchronize training dependency versions between Docker image and Python wheel. (#5261) 2020-09-23 19:03:42 -07:00
install_protobuf.sh Update protobuf to the latest version (#7396) 2021-04-21 10:30:06 -07:00
install_server_deps.sh Update protobuf to the latest version (#7396) 2021-04-21 10:30:06 -07:00
install_ubuntu.sh Update CI build docker images (#7386) 2021-04-21 13:18:34 -07:00
requirements.txt pick onnx release candidate (#7177) 2021-04-22 23:57:09 -07:00