Install torchbench before running the bisection

pytorch · Mar 23, 2024 · bef7c75 · bef7c75
1 parent 6e93e00
commit bef7c75
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/.github/workflows/userbenchmark-a100-bisection.yml b/.github/workflows/userbenchmark-a100-bisection.yml
@@ -58,9 +58,6 @@ jobs:
           sudo nvidia-smi -pm 1
           sudo nvidia-smi -ac 1215,1410
           nvidia-smi
-      - name: Install Deps
-        run: |
-          sudo apt-get -y update && sudo apt -y update
       - name: Setup conda env
         run: |
           CONDA_ENV=${BASE_CONDA_ENV} . "${SETUP_SCRIPT}"
@@ -82,6 +79,8 @@ jobs:
           . "${SETUP_SCRIPT}"; cd benchmark
           python utils/cuda_utils.py --install-torch-build-deps
           python utils/cuda_utils.py --install-torchbench-deps
+          # Install torchbench
+          python install.py
       - name: Bisection
         run: |
           . "${SETUP_SCRIPT}"; cd benchmark