onnxruntime
Add CI pipeline to publish Python training package targeting Rocm
#7417
Merged

Add CI pipeline to publish Python training package targeting Rocm #7417

suffiank merged 21 commits into master from sukha/rocm-training-whl
suffiank
first attempt rocm training wheel
6c2e26cd
modifications needed to python packaging pipeline for Rocm 4.1
491c4fe4
changges to not conflict with cuda
63b41505
try adding Rocm tests as well
d7ee480b
try with tests in place
e3cd4f22
fix trailing ws
a39f0583
add training data
776b9061
try again as root for tests
a4a0fd1e
use python3
c373a809
typo
acf881b3
try to map video, render group into container
47925e1f
try again
70e61f46
try again
ca954b5b
try to avoid yum error code
f7149484
suffiank suffiank requested a review from liqunfu liqunfu 4 years ago
suffiank suffiank requested a review 4 years ago
suffiank suffiank requested a review from weixingzhang weixingzhang 4 years ago
snnn
snnn commented on 2021-04-22
snnn
snnn commented on 2021-04-22
snnn
snnn commented on 2021-04-22
snnn
snnn commented on 2021-04-22
make UID 1001
9c724e19
try without yum downgrade
44975340
define rocm_version=None
47dec1ce
remove CUDA related comments for Rocm Dockerfile
32ab52bd
Dont pin nightly torch torchvision torchtext versions as they expire …
12306df5
missed requirements-rocm.txt from last commit
37e9de58
fix whitespace
9a2be6fb
snnn
snnn approved these changes on 2021-04-23
suffiank suffiank merged 7a3c1787 into master 4 years ago
suffiank suffiank deleted the sukha/rocm-training-whl branch 4 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone