Update build_GPU_Compute5.yml

This commit is contained in:
McCloudS
2025-02-03 19:50:22 -07:00
committed by GitHub
parent 5f4ce1aa58
commit 72adcd4515

View File

@@ -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