Skip to content

Commit

Permalink
fix on trtCudaVersion (#23616)
Browse files Browse the repository at this point in the history
### 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. -->
  • Loading branch information
yf711 authored Feb 8, 2025
1 parent 740e9ab commit 0274b7b
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ steps:
displayName: Set trtCudaVersion
- ${{ if and(eq(parameters.CudaVersion, '12.2'), eq(parameters.TrtVersion, '10.8.0.43')) }}:
- powershell: |
Write-Host "##vso[task.setvariable variable=trtCudaVersion;]12.6"
Write-Host "##vso[task.setvariable variable=trtCudaVersion;]12.8"
displayName: Set trtCudaVersion
- script: |
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,12 +29,12 @@ steps:
azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/cuda_sdk/v${{ parameters.SecondaryCUDAVersion }}" $(Agent.TempDirectory)
displayName: 'Download Secondary CUDA SDK v${{ parameters.SecondaryCUDAVersion }}'
- ${{ if eq(parameters.DownloadTRT, 'true') }}:
- powershell: |
azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/local/${{ parameters.win_trt_folder_cuda12 }}" $(Agent.TempDirectory)
displayName: 'Download ${{ parameters.win_trt_folder_cuda12 }}'
- powershell: |
azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/local/${{ parameters.win_trt_folder_cuda11 }}" $(Agent.TempDirectory)
displayName: 'Download ${{ parameters.win_trt_folder_cuda11 }}'
- powershell: |
azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/local/${{ parameters.win_trt_folder_cuda12 }}" $(Agent.TempDirectory)
displayName: 'Download ${{ variables.win_trt_folder_cuda12 }}'
- task: BatchScript@1
displayName: 'setup env'
Expand Down

0 comments on commit 0274b7b

Please sign in to comment.