diff --git a/.github/workflows/benchmark_dispatch.yml b/.github/workflows/benchmark_dispatch.yml index 16dd1712f..85f4bbffa 100644 --- a/.github/workflows/benchmark_dispatch.yml +++ b/.github/workflows/benchmark_dispatch.yml @@ -40,8 +40,8 @@ jobs: with: runner: linux.aws.h100 python-version: "3.12" - image: nvidia/cuda:12.8.1-devel-ubuntu24.04 - runtime-version: cu128 + image: nvidia/cuda:13.0.1-devel-ubuntu24.04 + runtime-version: cu130 container-options: --gpus all alias: h100 num-shards: ${{ matrix.num_shards }}