[GitHub] Faldict edited a comment on issue #12142: Failed to import MXNet built with TensorRT

[GitBox <gi...@apache.org>]

Faldict edited a comment on issue #12142: Failed to import MXNet built with TensorRT
URL: https://github.com/apache/incubator-mxnet/issues/12142#issuecomment-413095946
 
 
   @KellenSunderland I uninstalled protobuf 3.5.1 and rebuild the whole toolchain. At present, MXNet could be imported successfully. It seems that you should constrain the protobuf version strictly.
   
   Further more, I tried to run a tensorrt baseline. I used the test code `incubator-mxnet/tests/python/tensorrt/test_tensorrt_lenet5.py` but got an unexpected error:
   
   ```
   “python3 test_tensorrt_lenet5.py” terminated by signal SIGSEGV (Address boundary error)
   ``` 
   
   As I set some breakpoints, I found this error occurs when executing this line:
   
   ```
           executor = mx.contrib.tensorrt.tensorrt_bind(sym, ctx=mx.gpu(0), all_params=all_params,
                                                        data=data_size,
                                                        softmax_label=(batch_size,),
                                                        grad_req='null',
                                                        force_rebind=True)
   
   ```
   
   where the symbol and parameters are trained by running `python3 lenet5_train.py`. So how to solve this problem?
   
   EDIT: when I dig deeper, the error occurs during the execution of `_LIB.MXExecutorSimpleBind()`.

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services