saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-14 20:48:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix on trtCudaVersion (#23616)

Browse source 
### 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. -->
This commit is contained in:
Yifan Li 2025-02-08 14:20:00 -08:00 committed by GitHub
parent 740e9ab9f8
commit 0274b7b82f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -44,7 +44,7 @@ steps:
displayName: Set trtCudaVersion displayName: Set trtCudaVersion
- ${{ if and(eq(parameters.CudaVersion, '12.2'), eq(parameters.TrtVersion, '10.8.0.43')) }}: - ${{ if and(eq(parameters.CudaVersion, '12.2'), eq(parameters.TrtVersion, '10.8.0.43')) }}:
- powershell: | - powershell: |
Write-Host "##vso[task.setvariable variable=trtCudaVersion;]12.6" Write-Host "##vso[task.setvariable variable=trtCudaVersion;]12.8"
displayName: Set trtCudaVersion displayName: Set trtCudaVersion
- script: | - script: |

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

@ -29,12 +29,12 @@ steps:
azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/cuda_sdk/v${{ parameters.SecondaryCUDAVersion }}" $(Agent.TempDirectory) 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 }}' displayName: 'Download Secondary CUDA SDK v${{ parameters.SecondaryCUDAVersion }}'
- ${{ if eq(parameters.DownloadTRT, 'true') }}: - ${{ 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: | - powershell: |
azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/local/${{ parameters.win_trt_folder_cuda11 }}" $(Agent.TempDirectory) 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 }}' 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 - task: BatchScript@1
displayName: 'setup env' displayName: 'setup env'