U盘PE| w764位旗舰版下载 | U盘装win7系统 | U盘启动 |win7pe | win10下载 |加入收藏土豆PE官网U盘PE,U盘装win7系统,win7pe,U盘启动,U盘装系统,w764位旗舰版下载站!
当前位置:主页 > 帮助中心 > 帮助中心 >

Windows10 安装TensorRT 在python使用

来源:http://www.tudoupe.com时间:2022-04-22

文章目录

  • 前言 必读!
  • 一、依赖
    • 1.更新显卡驱动
    • 来两杯鸡尾酒和泡菜
  • 二、安装CUDA
        • 1.下载
        • 2.安装
  • 三、安装cuDNN
    • 1.下载并解压
    • 2.复制
    • 3. 添加环境变量
    • 四,保持 紧紧的图书馆。
  • 四、医院设备
    • 一号,天文感应器
    • 2.解压
    • 3.添加环境变量
    • 4.安装库
      • (1)激活环境
      • (2) 安装
    • 5.测试
  • 五. 安装有单一命令的TensorRT 安装:仅适用于Python
    • 安装 ARORRT Python 时安装
    • 2.验证
  • 安装作为附件的 Pycuda 。
      • 1.直接pip安装
      • 二. 手动设置(首选)
      • 3. 测试

前言 必读!

安装过程以可见卡为基础,并选择适当版本。

版本选择流程:

  • 显卡->驱动->cuda
    • cuda-cuDNN
    • cuda->TensorRT

一个可见的卡选择驱动器驱动由兼容的驱动程序下载的 Cuda 驱动器。 Cuda 选择由下载的 TensorRT 检查支持的 CDNN 版本和 TensorRT 版本 。

你熟悉程序 准备开始安装

安装 ARORRT 之前, 您必须更新 Nvidia 驱动程序并安装所需的 CUDA 。

一、依赖

  • Nvidia驱动
  • cuda
  • cudnn
  • python

1.更新显卡驱动

下载 Nvidia 驱动程序

产品线,如果是笔记本,请选择笔记本

下载类型 :
游戏准备时可以使用补丁和游戏支持 。
工作室驾驶员,保持高质量
如果您想要玩游戏, 请选择第一个选项。 如果您不玩游戏, 可以同时使用两个选项 。

来两杯鸡尾酒和泡菜

如果您安装 cuda 和 cudnn, 请检查下一个版本。 如果没有跳过检查步骤 。

一,看古达版本。

nvcc -V

双人观察抱抱路线

Linux

which nvcc

windows

set cuda

三个,看看Cudnn版本。

Linux请将路线更改为上面显示的路线 。

cat /usr/local/cuda/cudnn.h |grep CUDNN_MAJORR -A 2

Windos

(默认目录为 C: Program fileTNDIACUDNv10.) (x)

切换磁盘( 必要时) 。

C:> G:

更改到整合路径请将路线更改为上面显示的路线 。

G:> cd G:NVIDA GPU Computing ToolkitCUDAv10.1include

打开版本文件时查看版本 。

G:> cudnn_version.h

您的版本号, 从上到下。

版本为8.3.2

见Pytorch的版本。

二、安装CUDA

查看支持版本

支持版本与匹配版本或更低版本一起显示在右上角。

1.下载

注意! 不要下载最新版本。 也许没有匹配的 Cudnn 和 stronort 。 不要问我怎么知道。 是的, 那个傻瓜就是我, 我倒下了 。

请接受我的歉意, Nvidia。

2.安装

选择安装, 然后监视路径并查看已安装的组件 。不要安装驱动

三、安装cuDNN

1.下载并解压

2.复制

3. 添加环境变量

四,保持 紧紧的图书馆。

四、医院设备

一号,天文感应器

TensorRT 应根据您的 CUDA 版本下载 。

com/nvidia-tensort-download. com/nvidia- tensort- load. 校对:Soup

再次检查版本!

2.解压

3.添加环境变量

TensorRT 的 lib 文件夹路径应该添加到系统环境变量中 。

4.安装库

(1)激活环境

重新打开 CD 并打开您的环境 。

conda activate <env-name>

转到 TensorRT 文件夹的位置 。

cd xxxxx/xxxxx/TensorRT-x.x.x.x

查看文件

dir

在这些目录中, 我们需要图形外科、 笔记外科、 python、 uff python_heel 文件 。

(2) 安装

安装期间, 文件名会使用制表卡键自动完成 。

a..
cd python
dir# Python 版本由若干轮式文件支持。
python -V看看Python版本
这是Python版本 现有的 Pip 安装 shorrt -xxcp 环境。
其余部分将直接打包, 每个文件夹都包含一个 轮子文件 。

b. 在邮递目录中插入一个 whl 文件。
感应流必须手动安装 。

pip install uff/uff-0.6.9-py2.py3-none-any.whl

c. 保存图形外科目录中的 whl 文件 。

pip install graphsurgeon/graphsurgeon-x.x.x-py2.py3-none-any.whl

d. 插入onnx-graphesurgeon目录中的 whl 文件 。

pip install onnx-graphsurgeon/onnx-graphsurgeon-x.x.x-py2.py3-none-any.whl

5.测试

5 3 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 -

b. C/C++中的测试

稍后,一个小问题。

五. 安装有单一命令的TensorRT 安装:仅适用于Python

使用 Python 安装

你只需要安装 Python 和 Cuda 和 Cuda 并且Cudnn 自己下载

我不知道你在说什么。 Python 文件目前只支持 Python 3.6至3.9版本和CUDA 11版本。这不适用于任何其他的Python版本或CUDA版本。目前,只有Linux操作系统和x86-64 CPU结构得到支持。这些轮式文件预计将以第7中心或更高版本以及Ubuntu 18.04或更高版本提供。

请在安装之前更新安装工具,以保证安装正确(记得过渡环境)。
pip install --upgrade setuptools pip wheel

注意: 如果您没有 root 权限, 请使用 Sudo 来获取这些权限 。

安装 ARORRT Python 时安装

激活环境
conda activate your_env_name
安装
pip install -U nvidia-tensorrt --index-url https://pypi.ngc.nvidia.com

2.验证

如果显示下列信息,更新驱动程序可能不会更新。

安装作为附件的 Pycuda 。

如果你必须使用金刚石,

1.直接pip安装

二. 手动设置(首选)

前往海关数据自动化系统网站,获取适当的海关数据自动化系统轮式文件。

https://www.lfd.uci.edu/~gohlke/pythonlibs/#pycuda

Cuda 102是10版,cp39是Python 3.9,Win_amd64是64位操作系统。

cd 到下载文件夹, 然后使用 pip 激活环境安装 。

3. 测试

转到 ARORT 路径并输入样本以提供样本文件夹 。

他会下载训练设备, 完成训练,然后测试它。这需要一些时间。

Copyright © 2012-2014 Www.tudoupe.Com. 土豆启动 版权所有 意见建议:tdsky@tudoupe.com

土豆系统,土豆PE,win7系统下载,win7 64位旗舰版下载,u盘启动,u盘装系统,win10下载,win10正式版下载,win10 RTM正式版下载,win8下载,电脑蓝屏,IE11修复,网络受限,4K对齐,双系统,隐藏分区,系统安装不了,U盘装系统,笔记本装系统,台式机装系统,diskgenius运用,GHSOT装系统,U盘修复,U盘技巧,U盘速度,U盘不能格式化,U盘复制发生错误,U盘加密,U盘选购,开机黑屏,蓝屏,进不了系统,上不了网,打不开程序,点击无反应,系统设置,PE个性化,PE添加网络,PE维护系统

点击这里给我发消息