install cuda when needed

robandpdx-org · Mar 27, 2024 · 6de2370 · 6de2370
1 parent 3cc4dde
commit 6de2370
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/.github/actions/build_conda/action.yml b/.github/actions/build_conda/action.yml
@@ -50,6 +50,13 @@ runs:
       run: |
         conda install -y -q anaconda-client
         conda config --set anaconda_upload yes
+    - name: Install CUDA toolkit
+      if: inputs.cuda != ''
+      uses: Jimver/[email protected]
+      with:
+        cuda: ${{ inputs.cuda }}
+        use-github-cache: false
+        use-local-cache: false
     - name: Conda build (CPU)
       if: inputs.label == '' && inputs.cuda == ''
       shell: ${{ steps.choose_shell.outputs.shell }}