Windows10安装TensorRT及环境配置

环境安装:
参考

  1. nvidia驱动下载
    直接运行"466.47-desktop-win10-64bit-international-dch-whql"文件安装就好,开始让你选择临时解压路径,然后可以自定义选择安装路径。

  2. cuda10.2(或其他),下载

    cuda与显卡驱动匹配
    在这里插入图片描述

    安装:
    (1) 运行软件“cuda_10.2.89_441.22_win10”,开始让你选择临时解压路径,然后可以自定义各个组件(dev、doc、samples)选择安装路径。
    (2)安装完成后打开cmd,输入nvcc - V测试是否安装成功。

  3. cudnn下载(需要对应cuda版本)

    安装:
    直接解压后,将lib、include、bin下的文件分别复制到cuda安装路径下的dev文件夹下的lib、include、bin中。
    测试:
    cmd打开终端,切换到G:\nvidia\CUDA10.2\dev\extras\demo_suite(cudaPATH/dev/extras\demo_suite下,C盘切换到其他盘,输入G:),然后运行deviceQuery.exe 和 bandwidthTest.exe 来查看GPU的状态。
    在这里插入图片描述
    在这里插入图片描述

TensorRT安装

  1. 下载,选择上面对应cuda版本、cudnn版本的进行下载。cudnn可通过解压后的G:\nvidia\cuda\include\cudnn_version.h查看。我下载的是·TensorRT 7.2.1 for Windows10 and CUDA 10.2 ZIP package·对应的就是cuda10.2\cudnn8.0。
  2. 安装

    解压,并将lib添加至系统变量(或者复制lib中的复制dll文件到cuda安装目录中bin下)
    执行以下bin下的trtexec,看是否报错。

  3. 测试(一定要用vs??)
  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值