modricwang/Pytorch-Model-to-TensorRT

hello,guys!I use your progarm and get a problem

brucelee78 opened this issue · 1 comments

Screenshot from 2019-04-21 23-15-30
My environment:
TRT:4.0.1.6
opencv:3.4.2
CUDA: 9.0
CUDNN = 7.0.5

Thank you!

Hi, @brucelee78 Did you solve it? I met the same problem.