Fix Python Linux GPU package name (#5943)

Fix Python Linux GPU package name. I accidentally added "noopenmp" to it.
This commit is contained in:
Changming Sun 2020-11-25 17:46:11 -08:00 committed by GitHub
parent 27513d1fd7
commit 5fdd9f0fd2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 4 additions and 7 deletions

View file

@ -141,7 +141,7 @@ Currently the NNAPI execution provider is the only execution provider that has s
- this is located in `build/Windows/<config>/<config>/dist/<package name>.whl` on Windows, or `build/Linux/<config>/dist/<package name>.whl` on Linux.
- `<config>` is the value from the `--config` parameter from the build command (e.g. Release)
- the package name will differ based on your platform, python version, and build parameters
- e.g. `pip install -U build\Windows\Release\Release\dist\onnxruntime_noopenmp-1.5.2-cp37-cp37m-win_amd64.whl`
- e.g. `pip install -U build\Windows\Release\Release\dist\onnxruntime-1.5.2-cp37-cp37m-win_amd64.whl`
- Create an ORT format model by running `tools\python\convert_onnx_models_to_ort.py` as per the above instructions, with the addition of the `--use_nnapi` parameter
- the python package from your 'full' build with NNAPI enabled must be installed for `--use_nnapi` to be a valid option
- this will preserve all the nodes that can be assigned to NNAPI, as well as setup the ability to fallback to CPU execution if NNAPI is not available at runtime, or if NNAPI can not run all the nodes due to device limitations.

View file

@ -2010,10 +2010,6 @@ def main():
if args.build:
if args.build_wheel:
nightly_build = bool(os.getenv('NIGHTLY_BUILD') == '1')
wheel_name_suffix = args.wheel_name_suffix
if not args.use_openmp and wheel_name_suffix is None:
wheel_name_suffix = 'noopenmp'
build_python_wheel(
source_dir,
build_dir,
@ -2027,7 +2023,7 @@ def main():
args.use_acl,
args.use_armnn,
args.use_dml,
wheel_name_suffix,
args.wheel_name_suffix,
args.enable_training,
nightly_build=nightly_build,
featurizers_build=args.use_featurizers,

View file

@ -190,6 +190,7 @@ stages:
--enable_lto \
--build_wheel \
--enable_onnx_tests \
--wheel_name_suffix=noopenmp \
${{ parameters.build_py_parameters }} \
--cmake_extra_defines PYTHON_INCLUDE_DIR=$(python.manylinux.include.dir) PYTHON_LIBRARY=/usr/lib64/librt.so
workingDirectory: $(Build.SourcesDirectory)

View file

@ -38,7 +38,7 @@ python3 /onnxruntime_src/tools/python/convert_onnx_models_to_ort.py \
find /home/onnxruntimedev/.test_data/ort_minimal_e2e_test_data -type f -name "*.onnx" -delete
# Uninstall the ORT python wheel
python3 -m pip uninstall -y onnxruntime_noopenmp
python3 -m pip uninstall -y onnxruntime
# Clear the build
rm -rf /build/Debug