TensorRT 配置

step1:英伟达官网下载TensorRT,并解压。

NVIDIA TensorRT | NVIDIA Developerhttps://developer.nvidia.com/zh-cn/tensorrtstep2:将TensorRT 的lib文件夹里的.dll 文件复制到CUDA 的bin文件夹中,比如我的     D:\TensorRT-8.2.4.2\lib 中的.dll 文件 复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin  中。

step3:将TensorRT的python文件夹中与Anaconda 所创建环境的python 版本一致的.whl文件安装到Anaconda 环境中。比如我的D:\TensorRT-8.2.4.2\python\tensorrt-8.2.4.2-cp36-none-win_amd64.whl 安装如下图

TensorRT 配置_第1张图片

至此,TensorRT 就配置好了。

你可能感兴趣的:(TensorRT,ai)