Windows6:CUDA、cuDNN、TensorRT版本

Windows6:CUDA、cuDNN、TensorRT版本

版本匹配

CUDA

找到CUDA对应版本,点击进去。点击【Release Notes】查看,这里以CUDA Toolkit v11.6.1为例
notion image

cuDNN

下载cuDNN后缀就是对应的TensorRT版本
notion image
 

TensorRT

版本对应cuDNN的版本,这里以TensorRT 8.4.1为例
notion image

常见报错解决

CMake编译YOLOV5报错:No CUDA toolset found

notion image
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\extras\visual_studio_integration\MSBuildExtensions 下的四个文件,拷贝到以下路径:C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\MSBuild\Microsoft\VC\v160\BuildCustomizations ,再编译,通过。
notion image
notion image

Could not locate zlibwapi.dll

notion image
下载
选择zlib123dll.zip.
notion image
解压文件
  • zlibwapi.lib 放在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\lib
  • zlibwapi.dll 放在 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\bin
notion image

Could not load library cudnn_cnn_infer64_8.dll.

TensorRT was linked against cuDNN 8.4.1 but loaded cuDNN 8.4.0

notion image
在安装了zlip之后仍然出现该问题,最有可能是安装的CUDNN版本与电脑支持的CUDNN版本不一致
可在https://developer.nvidia.com/rdp/cudnn-archive重新下载对应的版本。把里面的bin、lib、include文件复制放在cuDNN对应bin、lib、include目录里面。C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4

TensorRT was linked against cuBLAS/cuBLAS LT 11.8.0 but loaded cuBLAS/cuBLAS LT 11.6.5

这是因为安装的cuda是11.6.5版本的,里面的cublas不是11.8.0,安装11.8.0版本的cuda就好了,我们可以看一下cuda11.3.1的介绍
notion image
notion image
我当时安装的是CUDA11.4.4,我又查了一下11.4.4对应的cuBLAS,果然是11.6.5👉,Nvidia更新驱动CUDA,没有更新对应的组件😂,重新安装CUDA11.6.1
notion image
 
 

版本

CUDA
cuDNN
TensorRT
Visual Studio
cuBLAs
Torch
11.7/11.6
8.4.1
8.4.1
2022(11.7)
11.10.1.25
1.12.0