Windows 环境下TensorRT 7.2 安装过程

安装环境

Anaconda3 / Python 3.7
Tensorflow-gpu 2.4
TensorRT 7.2.16
CUDA 10.2
CUDNN 8.0.4
VS 2019

简介

TensorRT是一个高性能的深度学习推理(Inference)优化器,可以为深度学习应用提供低延迟、高吞吐率的部署推理。TensorRT可用于对超大规模数据中心、嵌入式平台或自动驾驶平台进行推理加速。TensorRT现已能支持TensorFlow、Caffe、Mxnet、Pytorch等几乎所有的深度学习框架,将TensorRT和NVIDIA的GPU结合起来,能在几乎所有的框架中进行快速和高效的部署推理。 详细信息各位大佬已经写的很清楚了:
高性能深度学习支持引擎实战——TensorRT

安装过程

1、 目前TensorRT的Windows zip包不提供Python支持。需要自行安装PyCUDA配合使用,在Anaconda下使用pip install pycuda进行安装。[1]
2、根据自己的CUDA与cudnn版本选择合适的TensorRT版本:
英伟达TensorRT 下载地址:下载链接
基于我的个人版本选择了TensorRT 7.2.1 for Windows10 and CUDA 10.2 ZIP package 文件。

  • 3
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
要在Windows终端上安装TensorRT,您需要按照以下步骤进行操作: 1. 首先,您需要下载与您的CUDA版本和CUDNN版本相对应的TensorRT安装包。您可以在NVIDIA的TensorRT页面上找到适合您的版本。请确保您已登录NVIDIA账号,以便获得下载选项。\[2\]\[3\] 2. 下载完成后,解压缩TensorRT安装包到您选择的目录。 3. 打开命令提示符(cmd)并导航到TensorRT安装目录。 4. 运行安装命令,根据您的需求选择安装组件和路径。例如,您可以使用以下命令安装TensorRT: ``` msiexec /i TensorRT-7.2.1.6.Windows10.x86_64.cuda-10.2.cudnn8.0.zip ``` 请注意,上述命令中的文件名和路径应根据您下载的TensorRT安装包进行相应更改。 5. 安装完成后,您可以在Windows终端中使用TensorRT。您可以通过运行以下命令来验证TensorRT是否成功安装: ``` nvcc -V ``` 如果您看到与您安装TensorRT版本相对应的输出信息,则表示TensorRT已成功安装。\[1\] 请注意,安装TensorRT之前,您需要先安装适当版本的CUDA和CUDNN。您可以参考相关文档或链接\[3\]中提供的详细步骤来安装CUDA和CUDNN。 希望这些信息对您有所帮助! #### 引用[.reference_title] - *1* *2* [Windows10安装TensorRT环境配置](https://blog.csdn.net/qq_32768679/article/details/117325494)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [详解Windows系统安装TensorRT](https://blog.csdn.net/KRISNAT/article/details/130789078)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值