windows下配置tensorrt心得
先安装vs2017
先安装vs2017,不然后面使用cuda的时候会提示找不到cuda编译器,这时就要重装cuda
安装CUDA11.1
从nvidia的cuda官网下载CUDA11.1文件:cuda_11.1.0_456.43_win10.exe,并按步骤运行安装,安装程序会自动向系统添加CUDA相关环境变量
安装CUDNN8.2.1
从nvidia的cudnn官网下载cudnn安装包:cudnn-11.3-windows-x64-v8.2.1.32.zip
cudnn安装包内主要是一些头文件和库文件,直接按内容添加到cuda目录下的对应文件夹即可。
安装TensorRT7.2.3
从tensorrt官网下载tensorrt安装包:TensorRT-7.2.3.4.Windows10.x86_64.cuda-11.1.cudnn8.1.zip
将该安装包内容解压到某路径,添加目录下的bin和lib文件夹到环境变量
存在的问题
无
已解决的问题
- cuda版本为11.2时,安装TensorRT7.2.3无法跑通TensorRT的Sample,提示缺少nvrtc64_111_0.dll,这里的111对应的是cuda版本11.1,因为安装的cuda版本为11.2所以找不到运行的相关动态链接库,最后重装cuda版本为11.1解决
- 注意tensorrt对cuda版本的要求比较严格,cuda版本对vs的版本要求比较严格,如果版本选的不对例如cuda11.2配vs2015可能会导致工程无法生成
- cuda版本推荐为11.1