默认已安装CUDA、CUDNN、对应版本的pytorch等环境(注意版本对应就好)
重点说明用cmake去加载cuda和tensorRt库。
TensorRT下载官网:
https://developer.nvidia.com/nvidia-tensorrt-download
下载8 GA版本(稳定版本)即可。
将下载的压缩包解压,然后放到自己合适的目录。如:C:\Program Files\TensorRT-8.5.1.7
将解压后目录中的lib绝对路径添加到环境变量。
将lib目录下的dll文件复制到CUDA目录:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin。
- 如果是VS2019,就简单很多,直接选择导入项目,然后加载sln文件,run就好了。
- 如果是vscode或clion这种cmake来搭建环境的,照着下面的cmakelist写就好
cmake_minimum_required(VERSION 3.25)
project(tensorRtTest)
set(CMAKE_CXX_STANDARD