1、下载
下载链接
https://developer.nvidia.com/nvidia-tensorrt-8x-download
2、解压
下载的是一个zip文件,需要解压
3、cd 到
TensorRT-8.5.2.2.Windows10.x86_64.cuda-11.8.cudnn8.6\TensorRT-8.5.2.2\python
4、可以看到如下内容
选择虚拟环境相应的whl文件
比如:我的虚拟换进安装的python是3.9版本的,所以我需要安装的是
tensorrt-8.5.2.2-cp39-none-win_amd64.whl
5、cd到
TensorRT-8.5.2.2.Windows10.x86_64.cuda-11.8.cudnn8.6\TensorRT-8.5.2.2\lib
ctrl+a,全选之后,复制ctrl+c
6、将5的所有文件,复制到
NVIDIA GPU Computing Toolkit\CUDA\v11.6\bin
文件夹内
7、在anaconda相应的虚拟环境内输入如下内容