TensorRT环境配置(2023完整版)

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

配置TensorRT环境需要配置CUDA、CUDNN、英伟达显卡驱动

  • 安装顺序是先英伟达显卡驱动,CUDA,CUDNN,如果已经安装就可以跳过

可以用命令查询CUDA和英伟达驱动版本,下载对应的CUDNN:

nvidia-smi

类似这样,如图我的显卡驱动版本为515.65.01,CUDA版本为11.7

安装CUDNN

  1. 到官网下载对应CUDA版本的CUDNN,链接如下:

深度神经网络

下载需要英伟达账号:
  1. 下载对应的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
评论 1
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值