onnxruntime/orttraining/tools/ci_test
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
..
results change ci docker image to rocm6.1 (#21296) 2024-07-18 14:50:01 +08:00
compare_huggingface.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
compare_results.py [Better Engineering] Bump ruff to 0.0.278 and fix new lint errors (#16789) 2023-07-21 12:53:41 -07:00
download_azure_blob_archive.py Bump ruff in CI (#15533) 2023-04-17 10:11:44 -07:00
run_batch_size_test.py [Linter] Bump ruff and remove pylint (#17797) 2023-10-05 21:07:33 -07:00
run_bert_perf_test.py Bump ruff to 0.3.2 and black to 24 (#19878) 2024-03-13 10:00:32 -07:00
run_convergence_test.py [Linter] Bump ruff and remove pylint (#17797) 2023-10-05 21:07:33 -07:00
run_gpt2_perf_test.py [Linter] Bump ruff and remove pylint (#17797) 2023-10-05 21:07:33 -07:00