news 2026/10/21 2:55:19

ONNX Runtime、CUDA Toolkit 、cuDNN、驱动的安装

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
ONNX Runtime、CUDA Toolkit 、cuDNN、驱动的安装

ChatGLM3Loader执行时发生错误: Library cudart is not initialized。

文章《Chatglm3部署踩坑记录》里提到是因为没有安装 CUDA Toolkit 引起。

下面内容来自deepseek

关于NVIDIA驱动、CUDA Toolkit版本以及ComfyUI的依赖关系,以下是关键点解析:


1.nvidia-smi与nvcc -V的区别

  • nvidia-smi显示的CUDA版本(12.8)
    表示当前GPU驱动支持的最高CUDA Toolkit版本。这是驱动层的能力上限,但不意味着系统已安装此版本的CUDA Toolkit。驱动向下兼容,例如支持12.8的驱动可以运行12.x的所有CUDA程序。

  • nvcc -V显示的CUDA版本(12.4)
    表示当前系统中实际安装的CUDA Toolkit版本(即编译器版本)。这是开发环境实际使用的版本,决定了代码编译时的兼容性。


2.ComfyUI依赖的CUDA版本

ComfyUI的运行主要依赖于以下两个组件:

  • PyTorch的CUDA版本
    PyTorch预编译版本会绑定特定的CUDA Toolkit(如PyTorch 2.3.0可能对应CUDA 12.1)。通过以下命令查看PyTorch实际使用的CUDA版本:

    bash

    复制

    python -c "import torch; print(torch.version.cuda)"

    若输出为12.1,则即使系统安装了CUDA 12.4,PyTorch仍会使用其内置的12.1版本。

  • 系统CUDA Toolkit版本(nvcc -V)
    仅在以下情况相关:

    • 需要从源码编译PyTorch或其他CUDA扩展。

    • 使用依赖nvcc的第三方库(如自定义CUDA算子)。


3.版本兼容性建议

  • 驱动版本 ≥ CUDA Toolkit要求
    确保GPU驱动版本满足PyTorch或CUDA Toolkit的最低要求(例如CUDA 12.4要求驱动≥535.86.01)。

  • PyTorch与CUDA Toolkit解耦
    通常情况下,PyTorch的预编译版本无需系统安装完整CUDA Toolkit即可运行,因其自带必要的CUDA动态库。实测版本不一致能运行。版本不一致应该会影响编译module。


4.解决版本冲突的步骤

  1. 确认PyTorch的CUDA版本:

    python -c "import torch; print(torch.__version__); print(torch.version.cuda)"

  2. 检查驱动兼容性:
    访问 NVIDIA驱动文档 确认驱动版本是否支持PyTorch所需的CUDA版本。

  3. 必要时更新PyTorch或CUDA Toolkit:

    # 安装指定CUDA版本的PyTorch pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121


总结

  • ComfyUI的CUDA依赖以PyTorch版本为准,与nvcc -V的关联仅限于源码编译场景。

  • 只要驱动版本足够(≥12.8支持所有12.x),PyTorch可独立运行,无需强制对齐系统CUDA Toolkit版本。

建议优先通过PyTorch的预编译版本管理CUDA依赖,避免因手动安装多版本CUDA Toolkit导致环境冲突。

1、用 nvidia-smi.exe 命令查看显卡当前驱动程序版本

用 nvcc -V 命令查看:

2、NVIDIA CUDA 工具包发行说明,每个 CUDA 工具包版本中打包的开发 NVIDIA GPU 驱动程序的版本如下所示。

可以根据上表要求,安装更新的驱动程序:

3、从 CUDA Toolkit 存档 下载想要安装的版本(选择与所安装的Pytorch 相同的cuda版本):

不玩游戏,也不需要自动更新驱动,所以不安装 Geforce Experience ,安装驱动是需要的:

CUDA Visual Studio Integration
No supported version of Visual Studio was found. Some components of theCUDA Toolkit will not work properly. Please install ivisual Studid first to get the fullfunctionality.

Installed:
- Nsight Monitor
Not Installed:
- Nsight for Visual Studio 2022
Reason: VS2022 was not found
- Nsight for Visual Studio 2019
Reason: VS2019 was not found
- Nsight for Visual Studio 2017
Reason: VS2017 was not found
- Integrated Graphics Frame Debugger and Profiler
Reason: see https://developer.nvidia.com/nsight-vstools
- Integrated CUDA Profilers
Reason: see https://developer.nvidia.com/nsight-vstools

安装 CUDA Toolkit 后,ChatGLM3Loader加载运行正常。

后来运行InstantID Face Analysis时出错如下:

onnxruntime::python::CreateExecutionProviderInstance CUDA_PATH is set but CUDA wasnt able to be loaded. Please install the correct version of CUDA andcuDNN as mentioned in the GPU requirements page (https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requirements), make sure they're in the PATH, and that your GPU is supported.

根据错误信息中提供的网址:https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html(CUDA 执行提供程序要求),发现,我还得将 CUDA Toolkit 降回 11.8 。

所以安装CUDA Toolkit 12.4.0版本是合适的。

安装ORT

python.exe -m pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/

查看cuDNN版本 ,在Windows系统中,可以进入CUDA的安装目录查看文件,例如:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include\cudnn.h

NVIDIA cuDNN --> 参考文献 (支持产品)

cuDNN 8.9 安装指南

NVIDIA cuDNN 主页----》下载 cuDNN 文库----》以前版本的存档

有些cuDNN版本,在上面的图形下载界面并不提供,需要从官方 Tarball 仓库下载。

📦 使用 Tarball 压缩包(官方推荐)

这是官方为高级用户提供的安装方式,虽然需要手动操作,但能确保你获得最新的 9.27.0 版本。

  1. 访问官方 Tarball 仓库:直接打开 NVIDIA 的 cuDNN 文件下载服务器地址:
    https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/windows-x86_64/

  2. 下载对应版本:在这个目录中,寻找名称类似cudnn-windows-x86_64-9.27.0.x-archive.zip的文件进行下载。

  3. 手动安装:解压下载的 ZIP 文件,然后按照官方文档的指引,将文件复制到 cuDNN 的安装目录:(v9.x中x为具体版本号,如:9.27.0.42 【官方文档有说明】)

    • 将bin\cudnn*.dll复制到C:\Program Files\NVIDIA\CUDNN\v9.x\bin

    • 将include\cudnn*.h复制到C:\Program Files\NVIDIA\CUDNN\v9.x\include

    • 将lib\x64\cudnn*.lib复制到C:\Program Files\NVIDIA\CUDNN\v9.x\lib
      (请将v9.x替换为你实际使用的目录名)

双击运行:

选精简安装,会连适用cuda11.8的cuDNN也安装了,而是我不需要的,所以选择自定义安装:

从上面也发现安装目录是: C:\Program Files\NVIDIA\CUDNN\v9.1,从这里也可以知道自己目前安装的是什么版本,而不是网上搜索到的到什么CUDA的目录去查看cuDNN版本(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vX.X\bin)。

发现安装了2个版本:

查看已经安装的包 .\python_embeded\python.exe -m pip list

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/10/21 2:54:45

C# P2P通信实战:NAT穿透与UDP打洞实现点对点直连

简介:C#利用Socket实现客户端之间直接通信的完整示例工程,面向初学网络编程或需要完成Socket课程设计的开发者,重点演示基于TCP的面向连接套接字通信模型。程序包含服务端与多个客户端:服务端能够响应单个或任意多个客户端连接请求…

作者头像 李华
网站建设 2026/10/21 2:54:34

基于TCP/IP的展厅智能中控系统:场景指令队列与设备联调实战

简介:一套基于TCP/IP的展厅智能中控软件程序,适合展厅、多媒体展项及小型集成项目使用。采用可视化所见所得的编辑方式,通过鼠标拖曳即可完成界面布局,无需编程,降低了非技术人员的配置门槛。UI素材全部存放于userData…

作者头像 李华
网站建设 2026/10/21 2:54:20

AI专著生成工具实战:学术写作的选型、工作流与避坑指南

我见过太多写专著的人,卡在最难受的阶段:框架反复推翻,文献越查越乱,写了三章的稿子回头看又是另一种语气。学术专著这种动辄三五十万字的工程,最耗人的不是"写",而是把散落的想法、文献、数据组…

作者头像 李华
网站建设 2026/10/21 2:54:18

从搜索式问答到理解式助教:RAG知识库落地实践全解析

刚接手这个项目的时候,我其实有点低估了它。内部知识库一直在用一套“搜索式问答”,效果嘛,能用,但离好用差得远。原话是“搜出来的东西看着相关,但没解决我的问题”。做了几个月的RAG落地之后,我才真正想明…

作者头像 李华
网站建设 2026/10/21 2:53:15

VMware解锁macOS全指南:unlocker原理、实操与避坑

简介:Unlocker 3.0.3是一款面向VMware用户的解锁工具,专用于解决macOS系统中文件被进程锁定而无法删除、移动或重命名的麻烦。该版本经实测与VMware 15.5.1完美兼容,使用时只需先暂停VMware相关进程,应用补丁后无需重启系统即可立…

作者头像 李华