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 路径并输入样本以提供样本文件夹 。
他会下载训练设备, 完成训练,然后测试它。这需要一些时间。
相关新闻
- 2022-08-04 WPF的由来
- 2022-08-04 Win11勒索软件防护怎么打开?Win11安
- 2022-08-04 Windows系统jdk的配置
- 2022-08-04 Windows10 OneNote怎么重新登录?如何重
- 2022-08-04 超好用的 Windows 效率工具推荐
- 2022-08-04 Windows如何在CMD或PowerShell中配置代理
- 2022-08-04 powershell和cmd对比
- 2022-08-04 【QT】Windows下QT下载安装
- 2022-08-04 windows下 C++ 实现类属性的get和set方
- 2022-08-04 Win11快速助手在哪里?Win11打开快速
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
