Skip to content

Commit 0274b7b

Browse files
authored
fix on trtCudaVersion (#23616)
### Description <!-- Describe your changes. --> TensorRT 10.8 zip file has suffix of cuda-12.8, not 12.6 ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. -->
1 parent 740e9ab commit 0274b7b

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

tools/ci_build/github/azure-pipelines/templates/jobs/download_win_gpu_library.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ steps:
4444
displayName: Set trtCudaVersion
4545
- ${{ if and(eq(parameters.CudaVersion, '12.2'), eq(parameters.TrtVersion, '10.8.0.43')) }}:
4646
- powershell: |
47-
Write-Host "##vso[task.setvariable variable=trtCudaVersion;]12.6"
47+
Write-Host "##vso[task.setvariable variable=trtCudaVersion;]12.8"
4848
displayName: Set trtCudaVersion
4949
5050
- script: |

tools/ci_build/github/azure-pipelines/templates/jobs/set-winenv.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,12 +29,12 @@ steps:
2929
azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/cuda_sdk/v${{ parameters.SecondaryCUDAVersion }}" $(Agent.TempDirectory)
3030
displayName: 'Download Secondary CUDA SDK v${{ parameters.SecondaryCUDAVersion }}'
3131
- ${{ if eq(parameters.DownloadTRT, 'true') }}:
32+
- powershell: |
33+
azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/local/${{ parameters.win_trt_folder_cuda12 }}" $(Agent.TempDirectory)
34+
displayName: 'Download ${{ parameters.win_trt_folder_cuda12 }}'
3235
- powershell: |
3336
azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/local/${{ parameters.win_trt_folder_cuda11 }}" $(Agent.TempDirectory)
3437
displayName: 'Download ${{ parameters.win_trt_folder_cuda11 }}'
35-
- powershell: |
36-
azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/local/${{ parameters.win_trt_folder_cuda12 }}" $(Agent.TempDirectory)
37-
displayName: 'Download ${{ variables.win_trt_folder_cuda12 }}'
3838
3939
- task: BatchScript@1
4040
displayName: 'setup env'

0 commit comments

Comments
 (0)