onnxruntime/tools/ci_build/github/azure-pipelines/orttraining-linux-gpu-ci-pipeline.yml
2020-03-11 14:41:32 -07:00

31 lines
906 B
YAML

trigger: none
jobs:
- template: templates/linux-ci.yml
parameters:
AgentPool : 'Linux-GPU-CUDA10'
JobName: 'Onnxruntime_Linux_GPU_Training'
SubmoduleCheckoutMode: 'recursive'
# TODO --skip_onnx_tests is used to disable onnx_test_runner test.
# onnx_test_runner test occasionally causes "CUDA failure 77: an illegal memory access was encountered"
# remove --skip_onnx_tests when the test failures are fixed.
BuildCommand: >
tools/ci_build/github/linux/run_dockerbuild.sh
-o ubuntu16.04 -d gpu -r $(Build.BinariesDirectory)
-x "
--enable_training
--config $(buildConfig)
--skip_onnx_tests
--build_wheel
"
DoNugetPack: 'false'
ArtifactName: 'drop-linux'
TimeoutInMinutes: 90
Strategy:
maxParallel: 2
matrix:
Debug:
buildConfig: Debug
Release:
buildConfig: Release