Disable H100 for now

triton-lang · Oct 18, 2023 · 61cfdd9 · 61cfdd9
1 parent 58d358f
commit 61cfdd9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/.github/workflows/torchbench.yml b/.github/workflows/torchbench.yml
@@ -19,7 +19,9 @@ jobs:
   TorchBench-Nvidia:
     strategy:
       matrix:
-        runner: [[self-hosted, A100], [self-hosted, H100]]
+        #runner: [[self-hosted, A100], [self-hosted, H100]]
+        # Disable H100 since we cna't sudo
+        runner: [self-hosted, A100]
 
     runs-on: ${{ matrix.runner }}
 
@@ -50,6 +52,8 @@ jobs:
           sudo apt-get install --yes libpango-1.0-0
           python3 -m pip install --upgrade pip
           python3 -m pip install lightning-utilities pyre_extensions
+          df -h
+          du --max-depth=1 -h
           python3 -m pip install --upgrade torchaudio==2.2.0.dev20231017 torch==2.2.0.dev20231017 torchrec==0.5.0.dev20231017 torchvision==0.17.0.dev20231017 --index-url https://download.pytorch.org/whl/nightly/cu121
           cd torchbench
           python3 install.py -v