Update build_GPU_Compute5.yml
This commit is contained in:
10
.github/workflows/build_GPU_Compute5.yml
vendored
10
.github/workflows/build_GPU_Compute5.yml
vendored
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user