【YoloX部署——配置cuda10.2 cudnn8.1 tensorrt7.2.3-onnx转trt】

本文详细介绍了如何在Windows系统上使用CUDA 10.2、CUDNN 8.1.0和TensorRT 7.2.3部署YOLOX模型。内容包括CUDA的安装配置、CUDNN的下载与配置、TensorRT的安装与测试,以及从ONNX模型转换为TensorRT模型的步骤。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

概要

将训练好的YOLOX模型部署在windows系统下的GPU上。这里我选用的tensorrt版本为7.2.3、cuda版本为10.2。现在也有更好的cuda版本和tensorrt版本,后续若有时间再进行其他版本的尝试。

cuda10.2安装配置

1、下载cuda10.2
打开官网(https://developer.nvidia.com/cuda-toolkit-archive),选择10.2下载。
在这里插入图片描述
2、运行安装包,路径默认,选择精简安装模式,后面一路默认就OK,安装成功关闭。
3、配置环境变量。
安装结束后,右键 我的电脑–>属性–>高级系统设置–>环境变量,系统变量中已经加入了cuda的两个路径。再在系统变量path中加入下面的4个变量.
在这里插入图片描述
在这里插入图片描述

在这里插入图片描述

<think>好的,我现在需要帮用户解决关于CUDA 11.1cuDNN 8.0.5和TensorRT版本选择的问题。首先,我得回忆一下这些库之间的版本兼容性关系。用户可能是在搭建深度学习环境时遇到了版本匹配的问题,比如在安装TensorFlow或PyTorch时需要这些依赖库的正确版本组合。 首先,用户提到的是CUDA 11.1。我知道CUDA的主版本号通常需要和cuDNN以及TensorRT的主版本号对应。但具体的次版本可能有不同的兼容性要求。接下来是cuDNN 8.0.5,这个版本应该对应CUDA 11.x系列。但需要确认cuDNN 8.0.5是否支持CUDA 11.1,或者是否需要更高版本的CUDA 11.x,比如11.0或11.2。 然后,TensorRT的版本选择是关键。TensorRT的版本必须与CUDAcuDNN的版本兼容。我需要查阅NVIDIA的官方文档或者版本支持矩阵。通常,每个TensorRT版本都会明确支持的CUDAcuDNN版本。例如,TensorRT 7.x对应CUDA 11.0,而TensorRT 8.x可能支持CUDA 11.1或更高。 可能存在的陷阱是,虽然CUDA 11.1是主版本,但cuDNNTensorRT可能有不同的次版本要求。比如,cuDNN 8.0.5可能需要CUDA 11.1的特定补丁版本,或者可能与CUDA 11.0兼容。这时候需要确认cuDNN 8.0.5的发布说明,看它支持哪些CUDA版本。 另外,用户可能没有明确说明他们使用的深度学习框架(如TensorFlow、PyTorch),这些框架对CUDAcuDNNTensorRT的版本也有要求。但用户的问题中未提及,所以可能需要假设他们只需要这三者的兼容性。 接下来,我需要验证几个点: 1. cuDNN 8.0.5是否支持CUDA 11.1?查资料发现,cuDNN的版本通常对应多个CUDA版本。例如,cuDNN 8.0.5可能支持CUDA 11.0到11.2。需要确认这一点。 2. 确定支持CUDA 11.1cuDNN 8.0.5的TensorRT版本。例如,TensorRT 7.2.3可能支持CUDA 11.1,而TensorRT 8.x可能需要更高版本的CUDA。 可能的解决步骤是: - 检查TensorRT的发布说明,寻找与CUDA 11.1cuDNN 8.0.5兼容的版本。 - 根据NVIDIA官方文档,找到对应的组合。 假设查证后发现,TensorRT 7.2.3支持CUDA 11.1cuDNN 8.0.5,而TensorRT 8.0.x可能要求更高版本的CUDA。但需要确认具体的小版本号。例如,TensorRT 8.0 GA可能支持CUDA 11.1,但需要查看具体文档。 另外,用户可能需要考虑是否必须使用CUDA 11.1,因为某些情况下,升级到更高版本的CUDA可能会获得更好的兼容性。但用户的问题明确要求基于现有的CUDA 11.1cuDNN 8.0.5选择TensorRT版本。 总结可能的答案:建议使用TensorRT 7.x的最新版本,如7.2.3,或者TensorRT 8.0.x中的某个版本,具体需参考官方文档的兼容性表格。例如,TensorRT 8.0.1.6可能支持CUDA 11.1cuDNN 8.0.5。需要提醒用户查阅官方文档确认,并提供相关链接或具体版本号。</think>针对 **CUDA 11.1 + cuDNN 8.0.5** 的环境,以下是 **TensorRT** 版本的推荐选择及详细分析: --- ### **1. 版本兼容性核心原则** - **CUDAcuDNNTensorRT 的版本需严格匹配**,否则可能导致编译/运行时错误。 - **TensorRTCUDA 的依赖更强**,需优先根据 CUDA 版本选择 TensorRT--- ### **2. 官方推荐组合** 根据 NVIDIA 官方文档: - **CUDA 11.1** 对应的 **TensorRT 版本为 7.2.3**(长期支持版本)。 - **cuDNN 8.0.5** 兼容 CUDA 11.1,且与 TensorRT 7.2.3 完全适配。 | 组件 | 推荐版本 | 验证来源 | |-----------|----------------|---------------------------| | CUDA | 11.1 | NVIDIA 官方文档 | | cuDNN | 8.0.5 | cuDNN 8.0.5 发行说明 | | TensorRT | 7.2.3.4 | TensorRT 7.2.3 发行说明 | --- ### **3. 替代方案(非官方推荐)** 若需使用 **TensorRT 8.x**,需注意: - **CUDA 11.1 需升级至 11.3**(因 TensorRT 8.x 要求 CUDA11.3)。 - **cuDNN 8.0.5 可保留**(需验证与 CUDA 11.3 的兼容性)。 --- ### **4. 操作步骤** 1. **安装 CUDA 11.1** 确保系统已正确安装驱动并配置环境变量: ```bash export PATH=/usr/local/cuda-11.1/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH ``` 2. **安装 cuDNN 8.0.5** 解压 cuDNN 包并复制文件至 CUDA 目录: ```bash tar -xzvf cudnn-11.1-linux-x64-v8.0.5.39.tgz sudo cp cuda/include/* /usr/local/cuda-11.1/include/ sudo cp cuda/lib64/* /usr/local/cuda-11.1/lib64/ ``` 3. **安装 TensorRT 7.2.3** 下载对应 DEB/TAR 包并按官方指南安装: ```bash sudo dpkg -i nv-tensorrt-repo-ubuntu1804-cuda11.1-trt7.2.3.4-ga-20210226_1-1_amd64.deb sudo apt-key add /var/nv-tensorrt-repo-cuda11.1-trt7.2.3.4-ga-20210226/7fa2af80.pub sudo apt update sudo apt install tensorrt ``` --- ### **5. 验证安装** - **检查 TensorRT 版本**: ```bash dpkg -l | grep TensorRT ``` - **运行示例代码**: ```python import tensorrt as trt print(trt.__version__) # 应输出 7.2.3.4 ``` --- ### **6. 注意事项** - **深度学习框架适配**:若使用 PyTorch/TensorFlow,需同步确认其对 CUDA 11.1 的支持(例如 TensorFlow 2.4+ 支持 CUDA 11.0-11.2)。 - **容器化部署**:建议使用 NVIDIA NGC 提供的预构建镜像(如 `nvcr.io/nvidia/tensorrt:21.03-py3`),避免手动配置依赖。 --- ### **总结** - **最佳组合**:CUDA 11.1 + cuDNN 8.0.5 + **TensorRT 7.2.3.4**。 - 若需更高性能或新特性,可升级至 **CUDA 11.3 + TensorRT 8.x**,但需重新验证兼容性。
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

wang151038606

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值