diff --git a/.github/workflows/notebook-pr.yaml b/.github/workflows/notebook-pr.yaml
index 5cbdfd03d..81e78eb2a 100644
--- 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