‘tensorrt.tensorrt.Builder‘ object has no attribute ‘build_cuda_engine‘
问题简述:AttributeError: 'tensorrt.tensorrt.Builder' object has no attribute 'build_cuda_engine'原因:tensorrt版本大于等于7时,不再使用build_cuda_engine。解决办法:print('[info] Building an engine...')if TRT_VERSION_MAJOR >= 7: # Create an optimization profile (se
原创
2022-01-13 20:04:04 ·
8160 阅读 ·
7 评论