配置TensorRT环境需要配置CUDA、CUDNN、英伟达显卡驱动
- 安装顺序是先英伟达显卡驱动,CUDA,CUDNN,如果已经安装就可以跳过
可以用命令查询CUDA和英伟达驱动版本,下载对应的CUDNN:
nvidia-smi
类似这样,如图我的显卡驱动版本为515.65.01,CUDA版本为11.7

安装CUDNN
- 到官网下载对应CUDA版本的CUDNN,链接如下:
下载需要英伟达账号:
- 下载对应的cudnn版本,这里附上tar版本的cudnn安装方式。这个版本适配CUDA11.x
tar -xvf cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz
cd cudnn-linux-x86_64-8.5.0.96_cuda11-archive/incl
配置TensorRT环境指南

配置TensorRT环境需要按照特定顺序安装NVIDIA驱动、CUDA、CUDNN,然后下载并安装TensorRT,最后配置Python环境和相关依赖。通过`nvidia-smi`命令检查版本,并从官方渠道下载兼容的CUDNN和TensorRT版本。安装完成后,可以使用YOLO的export脚本验证设置是否成功。
最低0.47元/天 解锁文章
5115





