TensorRT
文章平均质量分 71
tensorrt框架
wait a minutes
创作不易,可以订阅博主哦。
展开
-
【TensorRT】yolov5的pytorch模型转tensorrt模型(自己的数据集与模型)
代码参考: https://github.com/wang-xinyu/tensorrtx https://github.com/ultralytics/yolov5 1.环境 ubuntu16.04 python3.6 cuda10.1/cuda10.0 tensorrt7.0.0.11 2. 安装参考:https://github.com/wang-xinyu/tensorrtx/blob/master/tutorials/install.md tensorrt安装参考:https原创 2021-04-12 22:01:20 · 2413 阅读 · 3 评论 -
【TensorRT】ubuntu16.04+cuda10.1安装tensorrt6
参考:https://zhuanlan.zhihu.com/p/64053177 由于根据很多方法安装,可能这里借一点信息,那里借一点信息,所以最后综合写一下。 0.环境 ubuntu16.04 cuda10.1 cudnn7 tensorrt6.0.1.5 1.安装 1.1 下载 下载deb文件,https://developer.nvidia.com/nvidia-tensorrt-6x-download。 https://developer.nvidia.com/compute/.原创 2021-04-12 18:07:05 · 5112 阅读 · 0 评论