change to ubuntu container

NVIDIA · Dec 13, 2024 · e6db490 · e6db490
1 parent eeda2ad
commit e6db490
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/.github/workflows/gh-build-and-test.yml b/.github/workflows/gh-build-and-test.yml
@@ -84,12 +84,13 @@ jobs:
       contents: read  # This is required for actions/checkout
     runs-on: ${{ (inputs.host-platform == 'linux-x64' && 'linux-amd64-gpu-v100-latest-1') ||
                  (inputs.host-platform == 'linux-aarch64' && 'linux-arm64-gpu-a100-latest-1') }}
-    # # TODO: use a different (nvidia?) container, or just run on bare image
-    # container:
-    #   options: -u root --security-opt seccomp=unconfined --privileged --shm-size 16g
-    #   image: condaforge/miniforge3:latest
-    #   env:
-    #     NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
+    # Our self-hosted runners require a container
+    # TODO: use a different (nvidia?) container
+    container:
+      options: -u root --security-opt seccomp=unconfined --privileged --shm-size 16g
+      image: ubuntu:20.04
+      env:
+        NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
     needs:
       - build
     steps: