ci(cuda): install CUDA toolkit to build extension

huggingface · Apr 10, 2024 · 99c4cfc · 99c4cfc
1 parent 8ecc1d3
commit 99c4cfc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/.github/workflows/linux-cuda-tests.yml b/.github/workflows/linux-cuda-tests.yml
@@ -32,6 +32,10 @@ jobs:
 
     steps:
       - uses: actions/checkout@v2
+      - name: Install CUDA Toolkit
+        run: |
+          sudo apt-get update -y
+          sudo apt install nvidia-cuda-toolkit -y
       - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v2
         with: