mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-13 18:08:13 +00:00
24 lines
931 B
YAML
24 lines
931 B
YAML
resources:
|
|
pipelines:
|
|
- pipeline: build
|
|
source: 'Python CUDA ALT Packaging Pipeline'
|
|
trigger: true
|
|
branch: main # branch to pick the artifact, Used only for manual triggered pipeline runs for testing the pipeline itself
|
|
|
|
stages:
|
|
# ****The following Stage depend on all previous tags. ***
|
|
# GPU resources are very limited,
|
|
# To utilize gpu resource more efficiently, run GPU job only after all cpus jobs succeed
|
|
- stage: Linux_Test_CUDA_Alt_x86_64_stage
|
|
dependsOn:
|
|
jobs:
|
|
- template: templates/py-packaging-linux-test-cuda.yml
|
|
parameters:
|
|
arch: 'x86_64'
|
|
machine_pool: 'Onnxruntime-Linux-GPU'
|
|
python_wheel_suffix: '_gpu'
|
|
timeout: 480
|
|
docker_base_image: onnxruntimebuildcache.azurecr.io/internal/azureml/onnxruntime/build/cuda11_x64_almalinux8_gcc11:20241111.1
|
|
trt_version: '10.6.0.26-1.cuda11.8'
|
|
cuda_version: '11.8'
|
|
|