使用tensorrt 加速Ultra-Fast-Lane-Detection项目
运行环境:
项目 | 主机版本(on ubuntu 18.04) | AGX版本(Jetson 4.3) |
---|---|---|
cuda 10.0 | 10.0.130 | - |
cudnn7 | 7.6.5.32 | - |
tensorrt | 7.0.0.11-1+cuda10.0 | - |
pytorch | 1.4.0+cu100 | 1.2.0 |
torchvision | 0.5.0+cu100 | 0.4.0 |
项目地址:https://gitee.com/n__n/Ultra-Fast-Lane-Detection.git
pth 转 onnx
#!/usr