Onnxruntime: mklml python wheel - No module named 'onnxruntime.capi'

[lei-Qiao]

When I build ort 0.4.0 wheel for mklml with python3.6, it can be built successfully, but when I import, it shows error.

onnxruntime-0.4.0-cp36-cp36m-linux_x86_64.whl
Python 3.6.9 (default, Jul 13 2019, 14:51:44)
[GCC 8.3.0] on linux
Type "help", "copyright", "credits" or "license" for more information.

import onnxruntime
Traceback (most recent call last):
File "", line 1, in
File "/onnxruntime/onnxruntime/__init__.py", line 18, in
from onnxruntime.capi import onnxruntime_validation
ModuleNotFoundError: No module named 'onnxruntime.capi'

[jywu-msft]

I assume you installed the built wheel?
Looks like you are running python from the source directory so it's trying to import onnxruntime module frim onnxruntime subdirectory rather than the installed package location.
Please change to a different directory and try again.

[faxu]

@lei-Qiao - do you still need assistance on this?

[lei-Qiao]

Thanks for your help. This issue can be closed.