use the cuda toolkit image instead of the runtime?

Serapieum-of-alex · Aug 10, 2024 · b759c12 · b759c12
1 parent 0c1c624
commit b759c12
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -12,7 +12,7 @@ jobs:
     env:
       OS: ${{ matrix.os }}
     container:
-      image: pytorch/pytorch:2.4.0-cuda12.4-cudnn9-runtime
+      image: pytorch/pytorch:2.4.0-cuda11.8-cudnn9-devel
 
     steps:
     - name: Checkout repository