【TensorRT】Ubuntu+18.04TensorRT5.1.5.0+ONNX1.4.0+onnx_tensorrt 5.1环境安装 很详细

为了将pytorch模型转换为onnx,并用tensorrt加速,安装了TensorRT5.1.5+ONNX1.4.0+onnx_tensorrt 5.1 1.TensorRT功能说明 TensorRT是NVIDIA 推出的一款基于CUDA和cudnn的神经网络推断加速引擎,相比于一般的深度学习框架,在CPU或者GPU模式下其可提供10X乃至100X的加速,极大提高了深度学习模型在边缘设备上的推断
相关文章
相关标签/搜索