install nvidia toolkit

NeuromatchAcademy · Jul 9, 2024 · 49063b8 · 49063b8
1 parent 94bcfb0
commit 49063b8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/notebook-pr.yaml b/.github/workflows/notebook-pr.yaml
@@ -63,6 +63,7 @@ jobs:
       - name: Install dependencies
         if: "!contains(env.COMMIT_MESSAGE, 'skip ci') && contains(env.COMMIT_MESSAGE, 'ci:execute')"
         run: |
+          sudo apt install nvidia-cuda-toolkit -y
           sudo apt-get update && sudo apt install xvfb -y
           python -m pip install --upgrade pip wheel
           pip install -r requirements.txt