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