【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

相关推荐
qq_369224339 天前
Windows全系通用!ntdll.dll文件丢失、报错、闪退问题的完整排查与修复教程
windows·dll·dll修复·dll丢失·dll错误
阿米亚波9 天前
【Windows】QEMU 启动 openEuler aarch64/arm64 架构系统 + 离线软件源
linux·windows·经验分享·笔记·架构·arm
caimouse9 天前
Reactos 第 10 章 网络操作 — 10.3.1 NIC驱动
网络·windows
初圣魔门首席弟子9 天前
Node.js 详细介绍(知识库版)
windows·qt·node.js·知识库
CHENG-JustDoIt9 天前
AI工具 | 爆火开源项目Odysseus AI 工作台:从项目介绍、部署情况及其使用等多方位分析指南(含详细步骤)
大数据·人工智能·windows·python·ai·开源·github
kingbal9 天前
Windows:flutter环境搭建
windows·flutter
未若君雅裁9 天前
Python 数据容器详解,list、tuple、str、set、dict 到底怎么选
windows·python·list
CodeKwang9 天前
Windows 环境 OCCT 8.0 编译构建及与 Qt6 项目集成
windows·qt·opencascade
我是伪码农9 天前
小兔鲜1-25
linux·服务器·windows
vx-Biye_Design9 天前
springboot安阳地区研学旅游服务小程序-计算机毕业设计源码12785
java·vue.js·windows·spring boot·tomcat·maven·mybatis