From 6de23707989160f276d80d897775730d04ff9b25 Mon Sep 17 00:00:00 2001
From: Rob Anderson <robandpdx@github.com>
Date: Wed, 27 Mar 2024 16:13:28 -0600
Subject: [PATCH] install cuda when needed

---
 .github/actions/build_conda/action.yml | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/.github/actions/build_conda/action.yml b/.github/actions/build_conda/action.yml
index a4569282f0..9a933aff4d 100644
--- 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/cuda-toolkit@v0.2.14
+      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 }}