项目地址
https://github.com/shouxieai/tensorRT_cpp
cuda版本
- nvcc -V 查看当前生效的cuda版本,如果不是自己指定想要的,安装其他版本或切换到指定版本
cudnn 版本
- 找对应cuda的版本下载即可
tensorrt版本,
- 切换到自己的tensorrt版本,替换掉lean中的文件夹
- cuda 10.0,10.2对应一个版本
- cuda 11.0,11.1,11.3对应一个版本
整个项目需要修改的地方
- vcxproj中的cuda版本修改,如原来cuda10.0,切换到你的cuda版本cuda10.2,直接修改对应数字即可
- 配置vc++目录,包含目录和库目录修改即可,主要是修改的cuda路径和tensorrt 版本的路径
- 配置环境,调试->环境,修改cuda路径等
编译运行后注意的地方
- dll文件
- 将
cuda文件夹的dll文件,放到workspace下 - 将
tensorrt依赖的dll文件,放到workspace下
- 将
- 编译运行的版本和修改路径的版本要一致,要么都是release,要么都是debug
- 可以切换cuda生成的目标机器32位 ->64位
- cuda sm数量,项目属性-》cuda c++ -> device->code generation 改成对应自己机器的位数,如1080ti ->sm 6.1,
- cuda 11.0时。如果cuda初始化失败,可以更新显卡驱动
相关版本的链接地址:
https://blog.csdn.net/weixin_41449637/article/details/120289151