From 72adcd4515a61081b500233f8c45fb642db3d359 Mon Sep 17 00:00:00 2001
From: McCloudS <64094529+McCloudS@users.noreply.github.com>
Date: Mon, 3 Feb 2025 19:50:22 -0700
Subject: [PATCH] Update build_GPU_Compute5.yml

---
 .github/workflows/build_GPU_Compute5.yml | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/.github/workflows/build_GPU_Compute5.yml b/.github/workflows/build_GPU_Compute5.yml
index b22d77d..9311847 100644
--- a/.github/workflows/build_GPU_Compute5.yml
+++ b/.github/workflows/build_GPU_Compute5.yml
@@ -38,6 +38,16 @@ jobs:
         with:
           username: ${{ secrets.DOCKERHUB_USERNAME }}
           password: ${{ secrets.DOCKERHUB_TOKEN }}
+          
+      - uses: Jimver/cuda-toolkit@v0.2.21
+        id: cuda-toolkit
+        with:
+          cuda: '12.5.0'
+
+      - run: echo "Installed cuda version is: ${{steps.cuda-toolkit.outputs.cuda}}"
+
+      - run: echo "Cuda install location: ${{steps.cuda-toolkit.outputs.CUDA_PATH}}"
+
 
       - name: Build and push CPU Docker image
         uses: docker/build-push-action@v5