Windows10下NVIDA CUDA,cuDNN和TensorRT安装教程

本文档详细介绍了如何检查GPU是否支持CUDA,以及如何一步步安装CUDA、cuDNN和TensorRT。首先,通过查看显卡控制面板确认CUDA支持。接着,从NVIDIA官网下载并安装适合系统的CUDA版本,并验证安装成功。然后,下载并安装cuDNN,将文件复制到CUDA相应目录,并设置环境变量。最后,下载TensorRT安装包,解压后配置环境变量,使用pip安装相关库,并通过测试确保TensorRT安装正确。

一. 查看自己的GPU版本是否支持cuda

  1. 打开显卡的控制面板,查看显卡是否支持cuda

    在这里插入图片描述

二 .安装CUDA

  1. 官网查看cuda版本与所需系统固件的对应关系
    在这里插入图片描述

  2. 从官网下载所需的CUDA Toolkit Archive,这里下载的版本如下:
    在这里插入图片描述

  3. 按默认安装下载好的安装包(只需要安装选项选择自定义,然后全选就可以)
    在这里插入图片描述

  4. 安装好后,打开anaconda的terminal输入nvcc -V,显示如下则说明安装成功
    在这里插入图片描述

三. 安装cuDNN

  1. 官网下载CUDNN,同前,注意版本号。下载地址:https://developer.nvidia.com/rdp/cudnn-download
    在这里插入图片描述

  2. 下载CUDNN后,复制粘贴cuDNN里面的下面三个文件到CUDA的相应同名文件

    • cudnn\cuda\bin => CUDA\v11.4\bin
    • cudnn\cuda\include => CUDA\v11.4\include
    • cudnn\lib\x64 => CUDA\v11.4\lib\x64
      在这里插入图片描述
  3. 在系统环境变量中添加如下环境变量
    在这里插入图片描述

四. TensorRT安装

官网安装教程: https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#installing-zip
在这里插入图片描述

  1. 官网下载好安装包:https://developer.nvidia.com/tensorrt.
    在这里插入图片描述

  2. 解压安装包,将文件放置到自己喜欢的位置
    在这里插入图片描述

  3. 添加系统环境变量
    在这里插入图片描述

  4. 将TensorRT文件lib目录下的*.dll全部拷贝到CUDA安装目录的bin文件夹下
    在这里插入图片描述

  5. 进入以下目录
    在这里插入图片描述

  6. 进入cmd,进入自己的conda安装目录,运行以下代码

    根据自己的python版本以及tensorrt版本运行以下代码

    python.exe -m pip install tensorrt-8.2.3.0-cp38-none-win_amd64.whl
    

    在这里插入图片描述

  7. 运行以下代码

    python3 -m pip install C:\Program Files\NVIDIA GPU Computing Toolkit\TensorRT-8.2.3.0\graphsurgeon\graphsurgeon-0.4.6-py2.py3-none-any.whl
    python3 -m pip install C:\Program Files\NVIDIA GPU Computing Toolkit\TensorRT-8.2.3.0\uff\uff-0.6.9-py2.py3-none-any.whl
    python3 -m pip install C:\Program Files\NVIDIA GPU Computing Toolkit\TensorRT-8.2.3.0\onnx_graphsurgeon\onnx_graphsurgeon-0.3.12-py2.py3-none-any.whl
    

    在这里插入图片描述

  8. 测试TensorRT是否安装成功,打开以下图片目录中的项目
    在这里插入图片描述

    1. 按以下操作更改项目使用的windows10 SDK至最新
      在这里插入图片描述
      在这里插入图片描述
  9. 运行程序
    在这里插入图片描述

  10. 出现以下结果说明安装成功
    在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

AoDeLuo

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值