Open Lamply opened 3 years ago
I met the same bug. Did you find out the solution?
I hit the same error when going through the example notebook at trt_pose (https://github.com/NVIDIA-AI-IOT/trt_pose/blob/master/tasks/human_pose/live_demo.ipynb)
I also tried your program and got the same issue. I'm on Ubuntu 18.04, python 3.6.9, CUDA 11.1 and TensorRT 7.2.3
Did anyone find a solution?
I met the same bug
@neoyang0620 @abaumhof @ygy12345678 FYI, Torch-TensorRT(https://github.com/NVIDIA/Torch-TensorRT) might be a better way to convert PyTorch models to TensorRT, or simply insert a dummy layer between them to avoid this issue.
想问一下后面这个问题是怎么解决的啊(卡好久了
Test code:
Log:
Environment: