@@ -19,7 +19,7 @@ concurrency:
1919jobs :
2020 build :
2121 name : Windows GPU TensorRT CI Pipeline
22- runs-on : ["self-hosted", "1ES.Pool=onnxruntime-github-vs2022-mms "]
22+ runs-on : ["self-hosted", "1ES.Pool=onnxruntime-github-vs2022-latest "]
2323 steps :
2424 - uses : actions/checkout@v5
2525 with :
@@ -41,10 +41,10 @@ jobs:
4141 working-directory : ${{ github.workspace }}
4242 shell : cmd
4343
44- - name : Download CUDA SDK v12.2
44+ - name : Download CUDA SDK v12.8
4545 working-directory : ${{ runner.temp }}
4646 run : |
47- azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/cuda_sdk/v12.2 " .
47+ azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/cuda_sdk/v12.8 " .
4848 dir
4949 shell : pwsh
5050
5656 shell : powershell
5757 run : |
5858 Write-Host "Adding CUDA to PATH"
59- Write-Host "CUDA Path: $env:RUNNER_TEMP\v12.2 \bin"
60- Add-Content -Path $env:GITHUB_PATH -Value "$env:RUNNER_TEMP\v12.2 \bin"
61- Add-Content -Path $env:GITHUB_PATH -Value "$env:RUNNER_TEMP\v12.2 \extras\CUPTI\lib64"
59+ Write-Host "CUDA Path: $env:RUNNER_TEMP\v12.8 \bin"
60+ Add-Content -Path $env:GITHUB_PATH -Value "$env:RUNNER_TEMP\v12.8 \bin"
61+ Add-Content -Path $env:GITHUB_PATH -Value "$env:RUNNER_TEMP\v12.8 \extras\CUPTI\lib64"
6262 Add-Content -Path $env:GITHUB_PATH -Value "$env:RUNNER_TEMP\TensorRT-10.9.0.34.Windows10.x86_64.cuda-12.8\lib"
6363
6464 - uses : actions/setup-node@v5
@@ -116,7 +116,7 @@ jobs:
116116 exit $lastExitCode
117117 }
118118 # Execute the build process
119- python ${{ github.workspace }}\tools\ci_build\build.py --config RelWithDebInfo --parallel --use_binskim_compliant_compile_flags --build_dir build --skip_submodule_sync --build_shared_lib --build --update --cmake_generator "Visual Studio 17 2022" --build_wheel --enable_onnx_tests --use_tensorrt --tensorrt_home="${{ runner.temp }}\TensorRT-10.9.0.34.Windows10.x86_64.cuda-12.8" --cuda_home="${{ runner.temp }}\v12.2 " --use_vcpkg --use_vcpkg_ms_internal_asset_cache --cmake_extra_defines CMAKE_CUDA_ARCHITECTURES=86
119+ python ${{ github.workspace }}\tools\ci_build\build.py --config RelWithDebInfo --parallel --use_binskim_compliant_compile_flags --build_dir build --skip_submodule_sync --build_shared_lib --build --update --cmake_generator "Visual Studio 17 2022" --build_wheel --enable_onnx_tests --use_tensorrt --tensorrt_home="${{ runner.temp }}\TensorRT-10.9.0.34.Windows10.x86_64.cuda-12.8" --cuda_home="${{ runner.temp }}\v12.8 " --use_vcpkg --use_vcpkg_ms_internal_asset_cache --cmake_extra_defines CMAKE_CUDA_ARCHITECTURES=86
120120 if ($lastExitCode -ne 0) {
121121 exit $lastExitCode
122122 }
@@ -193,10 +193,10 @@ jobs:
193193 working-directory : ${{ github.workspace }}
194194 shell : cmd
195195
196- - name : Download CUDA SDK v12.2
196+ - name : Download CUDA SDK v12.8
197197 working-directory : ${{ runner.temp }}
198198 run : |
199- azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/cuda_sdk/v12.2 " .
199+ azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/cuda_sdk/v12.8 " .
200200 dir
201201 shell : pwsh
202202
@@ -208,9 +208,9 @@ jobs:
208208 shell : powershell
209209 run : |
210210 Write-Host "Adding CUDA to PATH"
211- Write-Host "CUDA Path: $env:RUNNER_TEMP\v12.2 \bin"
212- Add-Content -Path $env:GITHUB_PATH -Value "$env:RUNNER_TEMP\v12.2 \bin"
213- Add-Content -Path $env:GITHUB_PATH -Value "$env:RUNNER_TEMP\v12.2 \extras\CUPTI\lib64"
211+ Write-Host "CUDA Path: $env:RUNNER_TEMP\v12.8 \bin"
212+ Add-Content -Path $env:GITHUB_PATH -Value "$env:RUNNER_TEMP\v12.8 \bin"
213+ Add-Content -Path $env:GITHUB_PATH -Value "$env:RUNNER_TEMP\v12.8 \extras\CUPTI\lib64"
214214 Add-Content -Path $env:GITHUB_PATH -Value "$env:RUNNER_TEMP\TensorRT-10.9.0.34.Windows10.x86_64.cuda-12.8\lib"
215215
216216 - name : Set OnnxRuntimeBuildDirectory
@@ -237,7 +237,7 @@ jobs:
237237 exit $lastExitCode
238238 }
239239
240- python ${{ github.workspace }}\tools\ci_build\build.py --config RelWithDebInfo --parallel --use_binskim_compliant_compile_flags --build_dir build --skip_submodule_sync --build_shared_lib --test --cmake_generator "Visual Studio 17 2022" --build_wheel --enable_onnx_tests --use_tensorrt --tensorrt_home="${{ runner.temp }}\TensorRT-10.9.0.34.Windows10.x86_64.cuda-12.8" --cuda_home="${{ runner.temp }}\v12.2 " --use_vcpkg --use_vcpkg_ms_internal_asset_cache --cmake_extra_defines CMAKE_CUDA_ARCHITECTURES=86
240+ python ${{ github.workspace }}\tools\ci_build\build.py --config RelWithDebInfo --parallel --use_binskim_compliant_compile_flags --build_dir build --skip_submodule_sync --build_shared_lib --test --cmake_generator "Visual Studio 17 2022" --build_wheel --enable_onnx_tests --use_tensorrt --tensorrt_home="${{ runner.temp }}\TensorRT-10.9.0.34.Windows10.x86_64.cuda-12.8" --cuda_home="${{ runner.temp }}\v12.8 " --use_vcpkg --use_vcpkg_ms_internal_asset_cache --cmake_extra_defines CMAKE_CUDA_ARCHITECTURES=86
241241 if ($lastExitCode -ne 0) {
242242 exit $lastExitCode
243243 }
0 commit comments