Only use network if cuda 11.8

tridao · Aug 10, 2023 · 78589ff · 78589ff
1 parent b91e513
commit 78589ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/build_wheels.yaml b/.github/workflows/build_wheels.yaml
@@ -67,7 +67,7 @@ jobs:
           cuda: ${{ matrix.cuda-version }}
           linux-local-args: '["--toolkit"]'
           # default method is "local", and we're hitting some error with caching for CUDA 11.8
-          method: 'network'
+          method: ${{ matrix.cuda-version == '11.8.0' && 'network' || 'local' }}
           # We need the cuda libraries (e.g. cuSparse, cuSolver) for compiling PyTorch extensions,
           # not just nvcc
           # sub-packages: '["nvcc"]'