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

CUDA

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

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

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

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

安装完成后会自动添加到系统的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即可安装。

相关推荐
70asunflower几秒前
CUDA编程指南基础知识点总结(5)
c++·人工智能·cuda
解道Jdon2 小时前
VSCode 2026年2月更新:技能市场、CLI、钩子支持
ide·windows·git·svn·eclipse·github·visual studio
love530love3 小时前
Windows 11 源码编译 vLLM 0.16 完全指南(RTX 3090 / CUDA 12.8 / PyTorch 2.7.1)
人工智能·pytorch·windows·python·深度学习·vllm·vs 2022
linuxxx1103 小时前
让openclaw使用系统命令:“rm“, “mkdir“, “touch“, “ls“, “cat“, “echo“
linux·服务器·windows
北极糊的狐3 小时前
Windows电脑能播放rmvb格式视频的软件,以及下载地址
windows·音视频
1nullptr3 小时前
Windows 10 文件名大小写敏感设置
windows
食指Shaye3 小时前
免费视频编辑软件FFmpeg 在windows上使用CMD命令行的操作说明
windows·ffmpeg
jason_renyu3 小时前
Windows 系统安装配置 MySQL 服务(含多端口/多实例管理)
windows·mysql·windows安装mysql·windows配置mysql·mysql多端口配置
娇娇yyyyyy3 小时前
QT编程(5):几种常用的对话框
windows·qt·microsoft
70asunflower4 小时前
CUDA基础知识巩固检验练习题【附有参考答案】(5)
人工智能·cuda·cpp