amend

pytorch · Nov 20, 2023 · 75959c7 · 75959c7
1 parent 053ea0d
commit 75959c7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/.github/workflows/benchmarks.yml b/.github/workflows/benchmarks.yml
@@ -53,8 +53,9 @@ jobs:
     defaults:
       run:
         shell: bash -l {0}
-    container: nvidia/cuda:12.3.0-base-ubuntu22.04
-    options: --gpus all
+    container:
+      image: nvidia/cuda:12.3.0-base-ubuntu22.04
+      options: --gpus all
     steps:
     - name: Install deps
       run: |