Environment:
GTX1080
Ubuntu 16.04
TensorRT 5.0.0.10
CUDA 9.0.176
If I follow the README.md to clone pybind11, install requirements and build the sample fc_plugin_caffe_mnist with following command:
cmake … -DPYBIND11_DIR=/home/xxx/yanyan/pybind11 -DNVINFER_LIB=/home/xxx/yanyan/TensorRT-5.0.0.10/lib/libnvinfer.so -DTRT_INC_DIR=/home/xxx/yanyan/TensorRT-5.0.0.10/include -DNVPARSERS_LIB=/home/xxx/yanyan/TensorRT-5.0.0.10/lib/libnvparsers.so -DNVINFER_PLUGIN_LIB=/home/xxx/yanyan/TensorRT-5.0.0.10/lib/libnvinfer_plugin.so -DCUDA_ROOT=/usr/local/cuda-9.0
make -j4
Then run python3 sample.py or python2 sample.py, I get an error:
Traceback (most recent call last):
File “sample.py”, line 12, in
from build import fcplugin
ImportError: generic_type: type “FCPluginFactory” referenced unknown base type “nvcaffeparser1::IPluginFactoryExt”
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File “sample.py”, line 17, in
Note that Python 2 requires the presence of __init__.py in the build folder"“”.format(err))
ImportError: ERROR: Failed to import module (generic_type: type “FCPluginFactory” referenced unknown base type “nvcaffeparser1::IPluginFactoryExt”)
Please build the FullyConnected sample plugin.
For more information, see the included README.md
Note that Python 2 requires the presence of __init__.py in the build folder