From 25db5706bb11c00b7244c5a951729cf4fcd160d9 Mon Sep 17 00:00:00 2001 From: Changming Sun Date: Mon, 28 Jun 2021 16:13:55 -0700 Subject: [PATCH] Change "Export PyTorch CustomOp" build pipeline to use Ubuntu 20.04 (#8158) Change "Export PyTorch CustomOp" build pipeline to use Ubuntu 20.04 --- .../azure-pipelines/linux-pytorch-custom-ops-ci-pipeline.yml | 4 ++-- tools/ci_build/github/linux/test_custom_ops_pytorch_export.sh | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tools/ci_build/github/azure-pipelines/linux-pytorch-custom-ops-ci-pipeline.yml b/tools/ci_build/github/azure-pipelines/linux-pytorch-custom-ops-ci-pipeline.yml index d5da2405ba..12de029b96 100644 --- a/tools/ci_build/github/azure-pipelines/linux-pytorch-custom-ops-ci-pipeline.yml +++ b/tools/ci_build/github/azure-pipelines/linux-pytorch-custom-ops-ci-pipeline.yml @@ -15,14 +15,14 @@ jobs: displayName: 'Docker build' inputs: script: | - docker build --pull -t onnxruntime-ubuntu --build-arg BUILD_USER=onnxruntimedev --build-arg BUILD_UID=$(id -u) --build-arg PYTHON_VERSION=3.6 -f Dockerfile.ubuntu . + docker build --pull -t onnxruntime-ubuntu --build-arg BUILD_USER=onnxruntimedev --build-arg BUILD_UID=$(id -u) --build-arg OS_VERSION=20.04 --build-arg PYTHON_VERSION=3.8 -f Dockerfile.ubuntu . workingDirectory: $(Build.SourcesDirectory)/tools/ci_build/github/linux/docker - task: CmdLine@2 displayName: 'ONNXRuntime build' inputs: script: | - docker run --rm --volume /data/onnx:/data/onnx:ro --volume $(Build.SourcesDirectory):/onnxruntime_src --volume $(Build.BinariesDirectory):/build --volume /data/models:/build/models:ro -e NIGHTLY_BUILD onnxruntime-ubuntu /usr/bin/python3.6 /onnxruntime_src/tools/ci_build/build.py --build_dir /build --config Release --build_shared_lib --cmake_path /usr/bin/cmake --build_wheel + docker run --rm --volume /data/onnx:/data/onnx:ro --volume $(Build.SourcesDirectory):/onnxruntime_src --volume $(Build.BinariesDirectory):/build --volume /data/models:/build/models:ro -e NIGHTLY_BUILD onnxruntime-ubuntu /usr/bin/python3 /onnxruntime_src/tools/ci_build/build.py --build_dir /build --config Release --build_shared_lib --cmake_path /usr/bin/cmake --build_wheel workingDirectory: $(Build.SourcesDirectory) - task: CmdLine@2 diff --git a/tools/ci_build/github/linux/test_custom_ops_pytorch_export.sh b/tools/ci_build/github/linux/test_custom_ops_pytorch_export.sh index 660baeb3af..56f5ff9f9e 100755 --- a/tools/ci_build/github/linux/test_custom_ops_pytorch_export.sh +++ b/tools/ci_build/github/linux/test_custom_ops_pytorch_export.sh @@ -5,7 +5,7 @@ pip3 install --user --upgrade pip pip3 install --user numpy==1.19.0 torch pytest pip3 install --user /build/Release/dist/*.whl -export PYTHONPATH=/onnxruntime_src/tools:/usr/local/lib/python3.6/site-packages:$PYTHONPATH +export PYTHONPATH=/onnxruntime_src/tools:/usr/local/lib/python3.8/site-packages:$PYTHONPATH python3 -m pytest -v /onnxruntime_src/tools/test/test_custom_ops_pytorch_exporter.py || exit 1