mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-30 20:18:08 +00:00
Fix Python Linux GPU package name (#5943)
Fix Python Linux GPU package name. I accidentally added "noopenmp" to it.
This commit is contained in:
parent
27513d1fd7
commit
5fdd9f0fd2
4 changed files with 4 additions and 7 deletions
|
|
@ -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.
|
||||
|
|
|
|||
|
|
@ -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,
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in a new issue