【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

相关推荐
水饺编程5 小时前
第5章,[Win32 章节] :Polygon 函数和多边形填充模式
c语言·c++·windows·visual studio
百事牛科技5 小时前
文档不想再加密了?Word打开密码移除的两种方法
windows·word
软件资深者8 小时前
千千静听典藏修复版 使用教程:经典本地音乐播放器回归,歌词同步、皮肤皮肤自由换,音乐播放器怀旧新手 5 分钟上手(2026)
windows
captain3768 小时前
文件与IO(2)
java·开发语言·windows·java-ee
星栖与芯10 小时前
# STM32MP157+LiteOS-M GCC极简工程:启动源码适配+Windows Make终极避坑
windows·stm32·单片机
AI视觉网奇10 小时前
图片转矢量图 算法总结
windows
清水白石00811 小时前
Python dataclass 高级避坑指南:从 default_factory、frozen、slots 到哈希灾难
windows·python·哈希算法
kaixin_啊啊11 小时前
Windows版ChatGPT启动失败Unable-to-locate-Codex-CLI-binary解决方法
windows·chatgpt
零号栈帧11 小时前
Codex 手机控制台——AgentDeck
windows·cloudflare·codex·pwa·ai agent
时光慢煮11 小时前
开源鸿蒙PC上手体验:从 Windows 和 macOS 切换过来,我被惊艳到了
windows·开源·harmonyos