一、安装GCC 5.4+
https://blog.csdn.net/weixin_64064486/article/details/123940266
二、安装pycuda
https://blog.csdn.net/zong596568821xp/article/details/86077553/
更改的地方是,安装pycuda用的是:
pip install pycuda
三、安装TensorRT
1、TensorRT下载
进入官方网站:https://developer.nvidia.com/nvidia-tensorrt-8x-download
2、解压安装
下载得到 zip 压缩包,解压。
任意顺序完成以下几步:
复制TensorRT-8.5.3.1\bin中内容到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin
复制TensorRT的include文件夹到CUDA的include文件夹
复制TensorRT-8.4.3.1\lib文件夹中的lib文件到CUDA的lib文件夹,dll文件到CUDA的bin文件夹
使用pip install xxx.whl安装TensorRT-8.4.3.1文件夹中的
3、使用 python 检查是否安装成功
>>>import tensorrt
>>>tensorrt.__version__
注意添加一下环境变量path
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\lib
否者提示
FileNotFoundError: Could not find: nvinfer.dll. Is it on your PATH?
由于TensorRT8.2.18没有myelin64_1.lib,会报错:无法打开输入文件“myelin64_1.lib“,如果是没有该库文件的,删除myelin64_1.lib这一行