From 3eecee6b51b3935ac5556dc71ebd9629511fe90f Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Sat, 28 Sep 2024 15:47:07 +0200 Subject: [PATCH] use available --- .github/workflows/test_onnxruntime_gpu.yml | 2 +- .github/workflows/test_onnxruntime_training.yml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/test_onnxruntime_gpu.yml b/.github/workflows/test_onnxruntime_gpu.yml index 7b47870c43a..4160bdbdc10 100644 --- a/.github/workflows/test_onnxruntime_gpu.yml +++ b/.github/workflows/test_onnxruntime_gpu.yml @@ -29,7 +29,7 @@ jobs: }} runs-on: - group: aws-g5-4xlarge-plus + group: aws-g6-4xlarge-plus container: image: nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04 diff --git a/.github/workflows/test_onnxruntime_training.yml b/.github/workflows/test_onnxruntime_training.yml index ad0bc934c42..4b8ab20a1d3 100644 --- a/.github/workflows/test_onnxruntime_training.yml +++ b/.github/workflows/test_onnxruntime_training.yml @@ -29,7 +29,7 @@ jobs: }} runs-on: - group: aws-g5-4xlarge-plus + group: aws-g6-4xlarge-plus container: image: nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04