当前位置: 首页 > news >正文

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

CUDA

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

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

image

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

image

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

image

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

image

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)。

image

下载完成后,将解压缩包中的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

http://www.lryc.cn/news/232158.html

相关文章:

  • 【C++】【Opencv】cv::GaussianBlur、cv::filter2D()函数详解和示例
  • 【数据分享】1961—2022年全国范围的逐日降水栅格数据
  • [Linux] 网络文件共享服务
  • 二十七、W5100S/W5500+RP2040树莓派Pico<iperf 测速示例>
  • 非常经典的一道SQL报错注入题目[极客大挑战 2019]HardSQL 1(两种解法!)
  • 创建一个用户test且使用testtab表空间及testtemp临时表空间并授予其权限,密码随意
  • Linux网络应用层协议之http/https
  • LLM(四)| Chinese-LLaMA-Alpaca:包含中文 LLaMA 模型和经过指令微调的 Alpaca 大型模型
  • 虚拟机CentOS 8 重启后不能上网
  • 让AI拥有人类的价值观,和让AI拥有人类智能同样重要
  • 《C++避坑神器·十七》找到程序崩溃Bug的一个实用方法:dump调试
  • ROS stm32 CAN通信
  • 简单地聊一聊Spring Boot的构架
  • 【算法】复习搜索与图论
  • 【KCC@南京】KCC南京数字经济-开源行
  • 苍穹外卖-day11
  • git_07_协同开发
  • 对比国内主流开源 SQL 审核平台 Yearning vs Archery
  • Mistral 7B 比Llama 2更好的开源大模型 (三)
  • 关于 Git 你了解多少?
  • 关于Elasticsearch的自动补全、数据同步和集群,以下是相关的知识点
  • linux套接字-Socket
  • debian 修改镜像源为阿里云【详细步骤】
  • 从0到0.01入门React | 004.精选 React 面试题
  • Linux 本地zabbix结合内网穿透工具实现安全远程访问浏览器
  • 【以图会意】文件系统从外存到内存到用户空间
  • 一、交换配置
  • 验证码:EasyDL 机器学习识别与云码平台一站式识别
  • postgreSQL中的高速缓存
  • 我把MySQL运行在Docker上,差点完了……