onnxruntime/tools/ci_build/github/linux/docker
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
..
manylinux2014_build_scripts Add CI pipeline to publish Python training package targeting Rocm (#7417) 2021-04-23 17:22:31 -07:00
scripts Add CI pipeline to publish Python training package targeting Rocm (#7417) 2021-04-23 17:22:31 -07:00
cuda_manylinux2010.repo
cuda_manylinux2014.repo
Dockerfile.arm_yocto
Dockerfile.manylinux2014_cpu
Dockerfile.manylinux2014_cuda10_2
Dockerfile.manylinux2014_cuda11
Dockerfile.manylinux2014_gpu
Dockerfile.manylinux2014_rocm Add CI pipeline to publish Python training package targeting Rocm (#7417) 2021-04-23 17:22:31 -07:00
Dockerfile.ubuntu
Dockerfile.ubuntu_for_android
Dockerfile.ubuntu_for_arm
Dockerfile.ubuntu_gpu
Dockerfile.ubuntu_openvino
Dockerfile.ubuntu_server
Dockerfile.ubuntu_tensorrt