TensorRT – Windows下TensorRT下载与配置
1 TensorRT下载
官方网站下载地址:https://developer.nvidia.com/zh-cn/tensorrt?term=TensorRT
选择合适的TensorRT版本下载,这里以TensorRT 7为例
进入TensorRT 7下载页面,勾选I Agree To the Terms of the NVIDIA TensorRT License Agreement
点击勾选之后会出来TensorRT 7的可供下载的版本,点开某个具体版本,根据自己安装的CUDA版本选择具体的TensorRT 版本下载,比如电脑上安装的版本是CUDA 10.2,则下载TensorRT 7.2.3 for Windows 10 and CUDA 10.2 ZIP package,注意虽然这上面是写着Windows 10,但是我在Win 7上也可以用。
2 下载相应版本的cuDNN库
TensorRT 下载下来之后是一个zip包,以上述TensorRT 7.2.3为例,下载之后zip包名为TensorRT-7.2.3.4.Windows10.x86_64.cuda-10.2.cudnn8.1.zip
在这其中就说明了cuda版本为cuda10.2,cudnn版本为cudnn8.1,如果之前在配置CUDA版本时,cuDNN的版本不是对应的版本,需下载TensorRT对应的cuDNN版本库,
否则在运行时会报无法找到cuDNN动态库的错误。
3 TensorRT在实际项目中的配置
TensorRT本质上就是一个第三方库,第三方一般都包括include,lib,dll三个文件夹,在下载的TensorRT的文件夹下也包含include和lib文件夹
其中include包含了头文件,而lib文件夹下包含了lib和dll文件,所以可以按常规的第三方库的配置方式配置TensorRT,在Visual Studio中针对某个项目进行包含目录、库目录配置即可。
本文作者:StubbornHuang
版权声明:本文为站长原创文章,如果转载请注明原文链接!
原文标题:TensorRT – Windows下TensorRT下载与配置
原文链接:https://www.stubbornhuang.com/1738/
发布于:2021年10月05日 23:11:22
修改于:2023年06月26日 21:13:02
当前分类随机文章推荐
- TensorRT - Polygraphy工具的使用 阅读5687次,点赞0次
- TensorRT - 计算模型推理时间 阅读527次,点赞1次
- TensorRT - Using an engine plan file across different models of devices is not recommended and is likely to affect performance or even cause errors 阅读584次,点赞0次
- TensorRT - 基于TensorRT的多线程并发推理方案 阅读783次,点赞2次
- TensorRT - Invalid Node - TopK,This version of TensorRT only supports input K as an initializer 阅读807次,点赞0次
- TensorRT - 安装TensorRT工具Polygraphy 阅读5331次,点赞0次
- TensorRT - Windows下TensorRT下载与配置 阅读2015次,点赞0次
- TensorRT - 使用torch普通算子组合替代torch.einsum爱因斯坦求和约定算子的一般性方法 阅读3472次,点赞3次
- TensortRT - 转换模型出现Could not locate zlibwapi.dll. Please make sure it is in your library path!错误 阅读1177次,点赞0次
- TensorRT - 扩展TensorRT C++API的模型输入维度,增加Dims5,Dims6,Dims7,Dims8 阅读2134次,点赞0次
全站随机文章推荐
- TensorRT - Windows下TensorRT下载与配置 阅读2015次,点赞0次
- 资源分享 - 动手学深度学习(李沐 2020年05月08日)PDF下载 阅读2684次,点赞0次
- 计算几何与计算机图形学必读书单整理 阅读23070次,点赞20次
- 盗墓笔记到底讲了什么故事???-带你领会终极和它 阅读4233次,点赞1次
- 资源分享 - 光线跟踪算法技术 Ray Tracing form the Ground Up中文版PDF下载 阅读1550次,点赞0次
- 资源分享 - Ray Tracing - A Tool for All 英文高清PDF下载 阅读2051次,点赞0次
- C++ - const_cast, static_cast, dynamic_cast, reinterpret_cast四种cast转换的区别和使用 阅读102次,点赞0次
- 资源分享 - Computer Graphics Programming in OpenGL with C++, First Edition 英文高清PDF下载 阅读1731次,点赞0次
- C++ - 使用模板和智能指针构建一个双向链表工具类 阅读1289次,点赞0次
- 宝塔面板部署风铃发卡系统小白教程 阅读773次,点赞1次
评论
169