win11安装TensorRT(python、c++(cmake、vs2019)部署)

默认已安装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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值