【TensorRT】实际测试中有效对应的cuda版本和cudnn版本

官网下载:

https://developer.nvidia.com/nvidia-tensorrt-8x-download

cuDNN Archive | NVIDIA Developer

CUDA Toolkit Archive | NVIDIA Developer

实测1

显卡:Quadro P3200 (计算能力 6.1)

CUDA:11.3.1

cuda_11.3.1_465.89_win10.exe

【TensorRT】实际测试中有效对应的cuda版本和cudnn版本_第1张图片

cudnn:8.2.1.32

cudnn-11.3-windows-x64-v8.2.1.32.zip

TensorRT:8.2.2.1

tensorrt-8.2.2.1.windows10.x86_64.cuda-11.4.cudnn8.2.zip

实测2

显卡:Quadro T1000(计算能力 7.5)

CUDA:10.2.89

cuda_10.2.89_441.22_win10.exe

cuda10.2需要安装2个补丁:

https://developer.download.nvidia.com/compute/cuda/10.2/Prod/patches/1/cuda_10.2.1_win10.exe

https://developer.download.nvidia.com/compute/cuda/10.2/Prod/patches/2/cuda_10.2.2_win10.exe

【TensorRT】实际测试中有效对应的cuda版本和cudnn版本_第2张图片

cudnn:8.2.1.32

cudnn-10.2-windows10-x64-v8.2.1.32.zip

TensorRT:8.0.3.4

tensorrt-8.0.3.4.windows10.x86_64.cuda-10.2.cudnn8.2.zip

查看CUDA中各个部分的版本:

Release Notes :: CUDA Toolkit DocumentationThe Release Notes for the CUDA Toolkit.https://docs.nvidia.com/cuda/archive/11.3.1/cuda-toolkit-release-notes/index.html比如CUDA 11.3.1

【TensorRT】实际测试中有效对应的cuda版本和cudnn版本_第3张图片

有些版本的TensorRT可能要求cuBlas是11.5.1 ,所以需要查看。

不然可能会出现这个报错:[TRT] TensorRT was linked against cuBLAS/cuBLAS LT 11.5.1 but loaded cuBLAS/cuBLAS LT 11.4.2_cumtchw-CSDN博客

查看显卡的计算能力

CUDA GPU | NVIDIA Developer

【TensorRT】实际测试中有效对应的cuda版本和cudnn版本_第4张图片

你可能感兴趣的:(TensorRT,深度学习,pytorch,神经网络)