关于tensorRT中遇到的几个问题

关于tensorRT中遇到的几个问题

一、运行同一个程序,前后两次的运行结果不一样
例如运行tensorRT-SSD代码,同样的测试图片,检测出来的目标框会有轻微的抖动,但是目标还是能检测出来,不会影响精度。

2018-11-5
找到原因所在了,softmax层的实现有问题。
只要是程序运行出现随机现象,那么一定是有问题的。一个正常的程序运行,如果自己没有写随机代码,那么程序肯定是有问题的。有很大可能是某个变量在定义的时候没有初始化。

二、相同的代码,相同的模型,相同的测试图片,在不同的平台上结果不一样
我在pc上的gtx1080显卡运行tensorRT代码,全部调试通过之后移植到tx2平台上运行,却发现在tx2平台上的检测精度要比pc上低一些

同一份代码,如果保证是三方库是一样的,绝不会出现一样的结果。要么就是针对不同的平台,相同版本的三方库的实现也是不一样的

TensorRT 是 NVIDIA 开发的一款深度学习优化框架,用于加速机器学习模型推理的速度并降低资源消耗。选择合适的 TensorRT 版本通常取决于您的硬件环境、CUDA 和 cuDNN 的版本以及您希望支持的功能。 以下是几个常见步骤供您参考,以便安装正确的 TensorRT 版本: ### 安装前准备 1. **确认系统兼容性**:首先,确定您的操作系统是否支持 TensorRTTensorRT 支持 Linux 系统,并推荐使用最新版的操作系统,如 Ubuntu 18.04 或更高版本。 2. **检查 CUDA 和 cuDNN 版本**:TensorRT 需要与 CUDA 和 cuDNN 搭配工作。确保已安装最新的 CUDA 版本,并且下载了相应的 cuDNN。在 NVIDIA 的开发者网站上可以找到这些组件的最新版本。 3. **获取安装脚本**:访问 NVIDIA 的开发者心,在“TensorRT”页面找到针对您操作系统的安装指南。通常会提供一个脚本文件,如 `nvidia-container-toolkit` 或其他特定于您环境的安装方法。 ### 实际安装过程 1. **下载安装脚本**:按照官方文档指引,下载适用于您的操作系统和软件包管理器的安装脚本。 2. **运行安装命令**:通过终端运行安装脚本,一般情况下脚本会自动处理依赖库的安装。例如: ```bash sudo ./install-tensorrt.sh -c cuda-version -d cudnn-version ``` 其 `cuda-version` 和 `cudnn-version` 分别代表您所使用的 CUDA 和 cuDNN 版本。注意替换为您实际需要的版本编号。 3. **验证安装成功**:安装完成后,可以通过运行一些示例程序或查看 `/usr/local/nvidia/lib64` 目录下是否有 `libnvinfer.so.*` 文件来验证 TensorRT 是否已成功安装。 ### 相关问题: 1. **如何解决安装过程遇到问题?** 当遇到安装错误时,可以查阅安装日志文件(通常位于 `/var/log/install.log`),或者访问 NVIDIA 开发者论坛寻求帮助。 2. **TensorRT 是否有针对特定需求的版本推荐?** 对于特定的应用场景(如嵌入式设备、高性能计算等),可能需要考虑 TensorRT 的不同版本及其提供的特性和性能优化。 3. **更新 TensorRT 至最新版本的方法是什么?** 跟随官方发布的更新指南,通常是通过重新运行安装脚本来获取最新版本,同时确保 CUDA 和 cuDNN 的兼容性。 --- 请注意,上述信息基于当前标准流程编写,具体的步骤和命令可能会随着 TensorFlow 版本、NVIDIA 的更新而发生变化。因此,建议直接从 NVIDIA 的官方网站获取最准确、详细的安装指导。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值