win11安装tensorRt成功

1.安装cuda

查看电脑cuda版本:nvidia-smi
在这里插入图片描述

(我的是11.6)下载链接:https://developer.nvidia.com/cuda-11-6-0-download-archive?target_os=Windows&target_arch=x86_64&target_version=11&target_type=exe_local
(建议默认地址安装,若选择位置,安装好后文件夹会消失)
网页下载链接慢的,可以从百度网盘下载文件:
链接:https://pan.baidu.com/s/1BkBAHr16hJhcQFTohk6N-A?pwd=1234
提取码:1234
安装好的位置为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6
在这里插入图片描述

2.下载cudnn(8.2.1)

下载链接为:https://developer.nvidia.com/rdp/cudnn-archive
在这里插入图片描述

3.下载tensorRT(8.4.1)

下载链接为:https://developer.nvidia.com/nvidia-tensorrt-8x-download)
在这里插入图片描述

将cudnn和tensorRT文件夹下的bin/lib-x64/include下的文件分别复制到cuda文件夹下的bin/lib-x64/include下。(默认位置为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6)

安装至python环境:
tensorRT文件下的python文件下有五个python版本的whl
在这里插入图片描述

pip install tensorrt-8.4.1.5-cp37-none-win_amd64.whl #(选择你自己的python版本)

安装测试:

查看cuda、cudnn是否安装成功:nvcc-V
在这里插入图片描述
测试python调用是否成功:

import tensorrt as trt

if __name__ == "__main__":
    print(trt.__version__)
    print("hello trt!!")

测试成功显示:

8.4.1.5
hello trt!!

参考链接:https://zhuanlan.zhihu.com/p/520995962
https://blog.csdn.net/zhuguiqin1/article/details/123844835
https://blog.csdn.net/weixin_43134049/article/details/124752259

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值