search

open-mmlab / mmdetection

OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io
Apache License 2.0
29.36k stars 9.42k forks source link

Tensorrt error #5472

Closed AM97Y closed 3 years ago

AM97Y commented 3 years ago

TensorRT Version-8.0.0.3: pythorch 1.7.0 CUDA Version-11.0: Operating System + Version- Ubuntu 20.04: Python Version-3.7.

I am trying to install tensorrt, but when compiling it writes an error and removes mmcv. With onnx, everything was fine.

image

RunningLeon commented 3 years ago

@AverinaMasha Hi, we only tested the supported models with the below config:

All models above are tested with Pytorch==1.6.0, onnx==1.7.0 and TensorRT-7.2.1.6.Ubuntu-16.04.x86_64-gnu.cuda-10.2.cudnn8.0