diff --git a/setup.py b/setup.py index d56382ab0d..a146ab1725 100644 --- a/setup.py +++ b/setup.py @@ -582,6 +582,10 @@ if enable_training or enable_training_apis: else: # cpu version for documentation local_version = "+cpu" + else: + if not (cuda_version or rocm_version): + # Training CPU package for ADO feeds is called onnxruntime-training-cpu + package_name = "onnxruntime-training-cpu" if package_name == "onnxruntime-tvm": packages += ["onnxruntime.providers.tvm"] diff --git a/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-cpu.yml b/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-cpu.yml index 92bfcb59a5..bdb0cd61db 100644 --- a/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-cpu.yml +++ b/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-cpu.yml @@ -76,6 +76,8 @@ stages: --volume $HOME/.onnx:/home/onnxruntimedev/.onnx \ -e NIGHTLY_BUILD \ -e BUILD_BUILDNUMBER \ + -e ORT_DISABLE_PYTHON_PACKAGE_LOCAL_VERSION \ + -e DEFAULT_TRAINING_PACKAGE_DEVICE \ onnxruntimetrainingcpubuild \ $(PythonManylinuxDir)/bin/python3 /onnxruntime_src/tools/ci_build/build.py \ --build_dir /build --cmake_generator Ninja \