onnxruntime/tools/ci_build/github/pai
kailums 1b38c05544
change ci docker image to rocm6.1 (#21296)
### Description
<!-- Describe your changes. -->
There is a bug for kernel running on rocm6.0, so change ci docker image
to rocm6.1

For the torch installed in the docker image, change to rocm repo when it
is not 6.0 version.

### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
2024-07-18 14:50:01 +08:00
..
buildtrainingdoc.sh Fix ROCm wheels pipeline after changes to manylinux scripts (#8026) 2021-06-10 21:01:28 -07:00
orttraining-ci.yml ROCm EP for AMD GPU (#5480) 2020-10-29 17:13:04 -07:00
pai-excluded-tests.txt [ROCm] Update ROCm and MIGraphX CI to ROCm5.7 (#17834) 2023-10-09 10:29:11 +08:00
pai_clean_device.sh [ROCm] Optimize ROCm CI pipeline 2 (#16691) 2023-07-24 13:57:48 +08:00
pai_huggingface_bert_large_test.sh Update cmake to 3.27 and upgrade Linux CUDA docker files from CentOS7 to UBI8 (#16856) 2023-09-05 18:12:10 -07:00
pai_test_launcher.sh ROCm EP for AMD GPU (#5480) 2020-10-29 17:13:04 -07:00
rocm-ci-pipeline-env.Dockerfile change ci docker image to rocm6.1 (#21296) 2024-07-18 14:50:01 +08:00