tkDNN(tensorRT版的目标检测网络库)在windows平台的配置过程

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 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值