tkDNN
tkDNN是YOLO官方认可的一个TensorRT版本的YOLO,之前都是Linux版本,最近才刚刚更新了windows版本的试验版,但是说实话没有国内[tensorrtx](https://github.com/wang-xinyu/tensorrtx)和[yolo-tensorrt](https://github.com/enazoe/yolo-tensorrt)这两个版本的好用和代码易读,之前用的Linux版本,Linux版本的配置过程挺方便的,windows版本配置挺麻烦,很多坑,和很多的未知的错误,配置了两天,咨询了作者两次才配置完成。文章目录
前言
一、tkDNN是什么?
tkDNN是一个使用了cuDNN和tensorRT的深度学习库,专门为英伟达开发板设计的,所以windows版本的tkDNN是实验阶段。tkDNN的检测速度如下图所示:
二、使用步骤
1.tkDNN的依赖
目前tkDNN所需要的依赖是:
1.WINDOWS 10 1803 或者更高
2.CUDA 10.0 (推荐 CUDA 11.2 ,我用的是CUDA 10.2)
3.CUDNN 7.6 (推荐 CUDNN 8.1.1 ,我用的是CUDNN 7.6.5)
4.TENSORRT 6.0.1