From e70344e648501dd0888a56fce88d43cc20fc1990 Mon Sep 17 00:00:00 2001 From: Changming Sun Date: Tue, 9 Feb 2021 11:04:39 -0800 Subject: [PATCH] Fix training python packaging pipeline (#6613) In a previous PR, I set the docker file name to a wrong value. --- .../azure-pipelines/templates/py-packaging-stage.yml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tools/ci_build/github/azure-pipelines/templates/py-packaging-stage.yml b/tools/ci_build/github/azure-pipelines/templates/py-packaging-stage.yml index 7cf6dbf07b..e2a9752e5c 100644 --- a/tools/ci_build/github/azure-pipelines/templates/py-packaging-stage.yml +++ b/tools/ci_build/github/azure-pipelines/templates/py-packaging-stage.yml @@ -289,13 +289,13 @@ stages: - template: get-docker-image-steps.yml parameters: - Dockerfile: tools/ci_build/github/linux/docker/Dockerfile.manylinux2014_cuda11 + Dockerfile: tools/ci_build/github/linux/docker/Dockerfile.manylinux2014_gpu Context: tools/ci_build/github/linux/docker DockerBuildArgs: >- --build-arg PYTHON_VERSION=$(PythonVersion) --build-arg INSTALL_DEPS_EXTRA_ARGS=-t --build-arg BUILD_UID=$(id -u) - Repository: onnxruntimecuda11build + Repository: onnxruntimetraininggpubuild - task: CmdLine@2 inputs: @@ -310,7 +310,7 @@ stages: -e NVIDIA_VISIBLE_DEVICES=all \ -e NIGHTLY_BUILD \ -e BUILD_BUILDNUMBER \ - onnxruntimecuda11build \ + onnxruntimetraininggpubuild \ $(PythonManylinuxDir)/bin/python3 /onnxruntime_src/tools/ci_build/build.py \ --build_dir /build \ --config Release \ @@ -320,7 +320,7 @@ stages: --enable_onnx_tests \ ${{ parameters.build_py_parameters }} \ --cmake_extra_defines CMAKE_CUDA_HOST_COMPILER=/opt/rh/devtoolset-8/root/usr/bin/cc PYTHON_INCLUDE_DIR=$(PythonManylinuxIncludeDir) PYTHON_LIBRARY=/usr/lib64/librt.so \ - --use_cuda --cuda_version=11.0 --cuda_home=/usr/local/cuda-11.0 --cudnn_home=/usr/local/cuda-11.0 + --use_cuda --cuda_version=10.2 --cuda_home=/usr/local/cuda-10.2 --cudnn_home=/usr/local/cuda-10.2 workingDirectory: $(Build.SourcesDirectory) - task: CopyFiles@2