search

DerryHub / BEVFormer_tensorrt

BEVFormer inference on TensorRT, including INT8 Quantization and Custom TensorRT Plugins (float/half/half2/int8).
Apache License 2.0
425 stars 69 forks source link

Dockerfile half updated? #77

Open jeeprhymeallthetime opened 1 year ago

jeeprhymeallthetime commented 1 year ago

The dockerfile seems to be installing a mix of dependencies between Open MMLabs, Nvidia, PyTorch, and TensorRT. I'm struggling to reconcile them.

I try mmcv==1.5.0 but it forces mmcv-full==1.7.0 which fails on the unit testing. And finding the right CUDA version that also connects properly with TensorRT is a struggle, as there's a lot of overlap.

Is there a way to reconcile these?

DerryHub commented 11 months ago

You can refer to #78