experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix tflite python whl package pybind11 build fail

This commit is contained in:
Leslie-Fang 2020-04-26 03:48:51 +08:00
parent 64c32cf510
commit b7b0761566
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/lite/tools/pip_package/README.md
View File

@ -9,6 +9,7 @@ To build a binary wheel run this script:
```sh
sudo apt install swig libjpeg-dev zlib1g-dev python3-dev python3-numpy
pip install numpy pybind11
sh tensorflow/lite/tools/make/download_dependencies.sh
sh tensorflow/lite/tools/pip_package/build_pip_package.sh
```

6
tensorflow/lite/tools/pip_package/setup.py
View File

@ -33,6 +33,7 @@ import sysconfig
from distutils.command.build_ext import build_ext
import numpy
import pybind11
from setuptools import Extension
from setuptools import find_packages
@ -178,8 +179,9 @@ ext = Extension(
'pip_package'),
numpy.get_include(),
os.path.join(DOWNLOADS_DIR, 'flatbuffers', 'include'),
os.path.join(DOWNLOADS_DIR, 'absl')
] + get_pybind_include(),
os.path.join(DOWNLOADS_DIR, 'absl'),
pybind11.get_include()
],
libraries=[LIB_TFLITE],
library_dirs=[LIB_TFLITE_DIR])