这篇文章的前提是你已经安装好了cuda和cudnn。并且没有问题了。不会弄得看我的文章。
然后开始配置trt内容。
1 相关参数
cuda cuda_11.1.0_456.43_win10.exe
cudnn cudnn-11.1-windows-x64-v8.0.4.30
trt TensorRT-7.2.1.6.Windows10.x86_64.cuda-11.1.cudnn8.0.zip
MSVS 2019 社区版
2 添加环境变量
C:\TensorRT-7.2.1.6.Windows10.x86_64.cuda-11.1.cudnn8.0\TensorRT-7.2.1.6\lib
C:\TensorRT-7.2.1.6.Windows10.x86_64.cuda-11.1.cudnn8.0\TensorRT-7.2.1.6\bin
3 安装以来包
进到我指出的文件夹中,然后pip安装就行。

4 复制dll文件到cuda安装目录
复制lib下文件到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin 下面。

本文指导读者在已安装CUDA和cuDNN的基础上,如何在Windows 10环境下配置Visual Studio 2019 (MSVS2019) 和 TensorRT。主要步骤包括设置环境变量、安装依赖包、复制DLL文件到CUDA目录,并通过测试代码验证配置是否成功。
最低0.47元/天 解锁文章

1804

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



