Windows8:TensorRT安装

Windows8:TensorRT安装

说在最前面的

  • TensorRT版本需要与CUDA、cuDNN版本对应
  • 下载TensorRT后的数字对应cuDNN版本
  • 📢📢📢更新CUDA后,CUDA cuBLAS版本不会更新,会导致报错类似TensorRT was linked against cuBLAS/cuBLAS LT 11.8.0 but loaded cuBLAS/cuBLAS LT 11.6.5错误

TensorRT兼容性查看

下载

notion image
notion image
notion image
Tips📢📢📢:这里TensorRT后面的8.4.1.5对应的cuDNN8.4.1,今天安装的时候cuDNN Archive才到8.4.0🤣🤣🤣,安装前建议先看一下cuDNN的版本(一边不会出现这种情况,赶巧了🙉)
notion image

安装

notion image
将下载好的文件解压,我这里解压到D盘
notion image

复制文件

  1. D:\TensorRT-8.4.1.5\lib所有lib复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\lib\x64
  1. D:\TensorRT-8.4.1.5\lib下所有dll复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\bin下,
  1. D:\TensorRT-8.4.1.5\include下所有.h文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\include
notion image
notion image

添加环境变量

右键此电脑—>属性—>高级系统设置—>环境变量,找到Path变量,选中并点击新建
notion image
notion image
将TensorRT的lib目录添加到Path
notion image