checking gpu with nvidia-smi

NVIDIA-Merlin · Oct 12, 2023 · 49795c2 · 49795c2
1 parent 7e4558c
commit 49795c2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/gpu.yml b/.github/workflows/gpu.yml
@@ -17,13 +17,13 @@ jobs:
       env:
         NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
       options: --shm-size=1G
-        --gpus=all
     steps:
       - uses: actions/checkout@v3
         with:
           fetch-depth: 0
       - name: Run tests
         run: |
+          nvidia-smi
           pip install tox
           ref_type=${{ github.ref_type }}
           branch=main
@@ -44,13 +44,13 @@ jobs:
       env:
         NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
       options: --shm-size=1G
-        --gpus=all
     steps:
       - uses: actions/checkout@v3
         with:
           fetch-depth: 0
       - name: Run tests
         run: |
+          nvidia-smi
           pip install tox
           ref_type=${{ github.ref_type }}
           branch=main