使用TensorRT遇到的小问题

1、使用TensorRT生成.engine文件时报错:

TensorRT was linked against cuBLAS/cuBLAS LT 11.3.0 but loaded cuBLAS/cuBLAS LT 11.2.0

问题原因:CUDA版本不对

解决:使用的TensorRT版本为7.2.1.6支持的CUDA版本为11.0 update1即11.0.3,安装11.0.3即可

2、本机GPU为gtx 960M,目标机器GPU为P2000,程序在目标机器运行报错:

[E] [TRT] ../rtSafe/coreReadArchive.cpp (41) - Serialization Error in verifyHeader: 0 (Version tag does not match)

问题原因:猜测是显卡类型不一致导致目标机无法反序列化.engine文件

解决:在目标机器上编译生成.engine文件​​​

你可能感兴趣的:(深度学习,深度学习,c++,gpu)