官网下载:
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
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
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
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中各个部分的版本:
有些版本的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博客
查看显卡的计算能力
Original: https://blog.csdn.net/jin739738709/article/details/122340814
Author: JinSu_
Title: 【TensorRT】实际测试中有效对应的cuda版本和cudnn版本
原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/708357/
转载文章受原作者版权保护。转载请注明原作者出处!