Building and caching deepspeed

Paladinium · Jan 18, 2025 · 2cb0768 · 2cb0768
1 parent e30f84d
commit 2cb0768
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/deepspeed.yml b/.github/workflows/deepspeed.yml
@@ -16,7 +16,8 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
-      - uses: Jimver/[email protected]
+      - name: Install NVIDIA CUDA Toolkit
+        uses: Jimver/[email protected]
         id: cuda-toolkit
         with:
           cuda: '12.4.1'