【Windows 开发环境配置——NVIDIA 篇】CUDA、cuDNN、TensorRT 三件套安装

CUDA

CUDA Toolkit Archive下载相应版本的离线安装包,这里以11.7为例。

打开安装包,在安装选项 选择自定义 模式,点击下一步

自定义安装选项 中,仅选择CUDA 组件(其中Nsight 相关组件用于代码调试与性能分析),若未安装显卡驱动,选择NVIDIA GeForce Experience components 并点击下一步

可能会出现如下图提示,警告你未安装Visual Studio (可参考【Windows 开发环境配置------C++ 篇】VSCode+MSVC/MinGW/Clangd/LLDB+Xmake 安装),这里打勾点击下一步即可。

安装完成后会自动添加到系统的PATH环境变量,无需手动添加。

cuDNN

cuDNN Archive下载相应版本的安装包,这里以8.4.1.50为例。

下载完成后,将解压缩包中的文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\cuDNN\v8.4.1.50目录下。

C:\Program Files\NVIDIA GPU Computing Toolkit\cuDNN\v8.4.1.50\bin添加到系统的PATH环境变量。

Zlib

Zlib是cuDNN所需的数据压缩软件库。在ZLib网站下载32bit64bitzlib123dll.zip(一般为64bit)。

下载完成后,将解压缩包中的zlibwapi.dll文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\cuDNN\v8.4.1.50\bin目录下,zlibwapi.lib文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\cuDNN\v8.4.1.50\lib目录下。

TensorRT

NVIDIA TensorRT 8.x Download下载相应版本的安装包,这里以8.4.2.4为例。

下载完成后,将解压缩包中的文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\TensorRT\v8.4.2.4目录下。

C:\Program Files\NVIDIA GPU Computing Toolkit\TensorRT\v8.4.2.4\bin添加到系统的PATH环境变量。

TensorRT Python 包安装

在终端打开C:\Program Files\NVIDIA GPU Computing Toolkit\TensorRT\v8.4.2.4\python目录,以python 3.10为例,输入pip install tensorrt-8.4.2.4-cp310-none-win_amd64.whl即可安装。

ONNX GraphSurgeon 包安装

在终端打开C:\Program Files\NVIDIA GPU Computing Toolkit\TensorRT\v8.4.2.4\onnx_graphsurgeon目录,输入pip install onnx_graphsurgeon-0.3.12-py2.py3-none-any.whl即可安装。

文章转载自:laugh12321

原文链接:https://www.cnblogs.com/laugh12321/p/17830096.html

相关推荐
Kiri霧2 小时前
Rust模式匹配详解
开发语言·windows·rust
程序设计实验室4 小时前
使用命令行删除 Windows 网络映射驱动器
windows
用户31187945592186 小时前
Windows 电脑安装 XTerminal 1.25.1 x64 版(带安装包下载关键词)
windows
Logintern097 小时前
windows如何设置mongodb的副本集
数据库·windows·mongodb
Chandler2411 小时前
一图掌握 操作系统 核心要点
linux·windows·后端·系统
ajassi200011 小时前
开源 C# 快速开发(十七)进程--消息队列MSMQ
windows·开源·c#
Python私教12 小时前
5分钟上手 MongoDB:从零安装到第一条数据插入(Windows / macOS / Linux 全平台图解)
windows·mongodb·macos
Yana_Zeng13 小时前
win10安装spark3.1详细流程(小白用)
hadoop·windows·spark
sukalot14 小时前
windows显示驱动开发-浮点、围栏支持、资源管理
windows·驱动开发
Hello.Reader15 小时前
Flink State V2 实战从同步到异步的跃迁
网络·windows·flink