栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

WIN10安装配置TensorRT详解

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

WIN10安装配置TensorRT详解

WIN10安装配置TensorRT详解
    • @[TOC](WIN10安装配置TensorRT详解)
        • 查看本机适配的CUDA 版本
        • 下载CUDA安装包
        • 下载cuDNN安装包
        • 下载TensorRT安装包
        • 流程化安装软件包
          • (1)安装CUDA
          • (2)安装cuDNN
          • (3)安装TensorRT
          • (4)验证TensorRT是否安装成功

工欲善其事,必先利其器

查看本机适配的CUDA 版本

要想安装TensorRT必须要先安装CUDA和cudnn,那么首先需要去查看自己电脑的英伟达驱动程序程序,位置如下:

NVIDIA控制面板->帮助->组件,如下图所示本机适配CUDA版本为11.6

下载CUDA安装包

进入CUDA 安装包的下载地址如下:

CUDA Toolkit 11.6 Downloads | NVIDIA Developer

按照机器适配版本,下载好安装包;如果没有帐户,需要先进行注册

下载cuDNN安装包

进入英伟达官方网站,选择和自己适配的cuDNN版本,地址如下:

cuDNN Download | NVIDIA Developer

如果没有帐户,需要先进行注册

下载TensorRT安装包

进入英伟达官网,选择和自己机器适配的TensorRT,地址如下:

NVIDIA TensorRT 8.x Download | NVIDIA Developer

流程化安装软件包 (1)安装CUDA

双击exe文件解压

NVIDIA安装->仅需一路点击确下一步然后结束

安装成功的标志:

(2)安装cuDNN

解压与拷贝

因cuDNN属于是CUDA的一个补丁包,因此不要安装,仅需要将上述三个binincludelib文件夹中的相关文件拷贝到CUDA安装位置的binincludelib文件夹中。

本电脑默认CUDA路径为:C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.6

验证

分别运行如下两个命令,若均为pass验证通过,则表示安装成功。

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.6extrasdemo_suite>bandwidthTest.exe

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.6extrasdemo_suite>deviceQuery.exe

(3)安装TensorRT

解压安装包后得到如下所示图:

安装TensorRT文件夹中的.whl文件*

pip install D:chorm-downloadTensorRT-8.4.1.5.Windows10.x86_64.cuda-11.6.cudnn8.4TensorRT-8.4.1.5pythontensorrt-8.4.1.5-cp38-none-win_amd64.whl 
其中cp38是依据环境使用的python版本获取的
pip install D:chorm-downloadTensorRT-8.4.1.5.Windows10.x86_64.cuda-11.6.cudnn8.4TensorRT-8.4.1.5graphsurgeongraphsurgeon-0.4.6-py2.py3-none-any.whl 
pip install D:chorm-downloadTensorRT-8.4.1.5.Windows10.x86_64.cuda-11.6.cudnn8.4TensorRT-8.4.1.5onnx_graphsurgeononnx_graphsurgeon-0.3.12-py2.py3-none-any.whl
pip install D:chorm-downloadTensorRT-8.4.1.5.Windows10.x86_64.cuda-11.6.cudnn8.4TensorRT-8.4.1.5uffuff-0.6.9-py2.py3-none-any.whl
(4)验证TensorRT是否安装成功

查看trt的版本

运行TensortRT文件夹下的sample.py

如果遇到错误:

解决方法:

去英伟达官网找到对应的zlibwapi.dll(依据电脑版本下载X64、X86版本的)文件,并将该dll拷贝到C:WindowsSystem32文件夹下

至此,windows安装TensorRT的过程就结束了。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/1001873.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号