search

shouxieai / tensorRT_Pro

C++ library based on tensorrt integration
MIT License
2.6k stars 549 forks source link

佬，之前使用yolox推导没有问题，然后现在突然报错:Tensor allocator query failed #131

Open wjyangqun opened 2 years ago

wjyangqun commented 2 years ago

佬，去年使用yolox推导没有问题，然后现在突然报错:Tensor allocator query failed. 而且我要是从加载onnx开始一套走下来，不会报错，但是单独加载引擎文件也会报上面的错:Tensor allocator query failed.

去年是用的老版的yolox训练然后来推导，今年使用较新版的yolox（归一化有区别 @`Y%HA5OG1655(79B}U@9`S ）

wjyangqun commented 2 years ago

还有，我用的3090ti会出现这个问题，1080ti不会

wjyangqun commented 2 years ago

3090ti权重只有100多m，1080ti有400m

wjyangqun commented 2 years ago

30系显卡加载第一张都超级慢，是显卡太新的原因么

wjyangqun commented 2 years ago

LTA)934@K}ZPS4V`7X0R(X7

shouxieai commented 2 years ago

这是显卡驱动、cudnn适配问题。一般容易出现在30系列上。你要保证你cudnn、显卡驱动都用比较新的