配置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
1万+



被折叠的 条评论
为什么被折叠?



