From 49063b80cc1372eddfe8708c4979da7cf9e36d2e Mon Sep 17 00:00:00 2001
From: Konstantine Tsafatinos <kosta.tsaf@gmail.com>
Date: Tue, 9 Jul 2024 16:17:20 -0400
Subject: [PATCH] install nvidia toolkit

---
 .github/workflows/notebook-pr.yaml | 1 +
 1 file changed, 1 insertion(+)

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