mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-14 20:48:00 +00:00
0274b7b82f
### 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. -->
44 lines
1.9 KiB
YAML
44 lines
1.9 KiB
YAML
parameters:
|
|
- name: EnvSetupScript
|
|
type: string
|
|
- name: DownloadCUDA
|
|
type: boolean
|
|
default: false
|
|
- name: DownloadTRT
|
|
type: boolean
|
|
default: false
|
|
- name: PrimaryCUDAVersion
|
|
type: string
|
|
default: '12.2'
|
|
- name: SecondaryCUDAVersion
|
|
type: string
|
|
default: '11.8'
|
|
- name: win_trt_folder_cuda11
|
|
type: string
|
|
default: 'TensorRT-10.8.0.43.Windows10.x86_64.cuda-11.8'
|
|
- name: win_trt_folder_cuda12
|
|
type: string
|
|
default: 'TensorRT-10.8.0.43.Windows10.x86_64.cuda-12.8'
|
|
|
|
steps:
|
|
- ${{ if eq(parameters.DownloadCUDA, 'true') }}:
|
|
- powershell: |
|
|
azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/cuda_sdk/v${{ parameters.PrimaryCUDAVersion }}" $(Agent.TempDirectory)
|
|
displayName: 'Download Primary CUDA SDK v${{ parameters.PrimaryCUDAVersion }}'
|
|
- powershell: |
|
|
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 }}'
|
|
|
|
- task: BatchScript@1
|
|
displayName: 'setup env'
|
|
inputs:
|
|
filename: '$(Build.SourcesDirectory)\tools\ci_build\github\windows\${{ parameters.EnvSetupScript }}'
|
|
modifyEnvironment: true
|
|
workingFolder: '$(Build.BinariesDirectory)'
|