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

【Stable Diffusion部署至GNU/Linux】安装流程

以下是安装Stable Diffusion的步骤,以Ubuntu 22.04 LTS为例子。

显卡与计算架构介绍

CUDA是NVIDIA GPU的专用并行计算架构
技术层级 说明
CUDA Toolkit 提供GPU编译器(nvcc)、数学库(cuBLAS)等开发工具
cuDNN 深度神经网络加速库(需单独下载)
GPU驱动 包含CUDA Driver(需与CUDA Toolkit版本匹配)

CUDA与NIDIA

  1. 硬件指令集绑定:NVIDIA GPU内置CUDA核心需要专用指令集
  2. 生态垄断地位:90%的深度学习框架原生支持CUDA(PyTorch/TensorFlow等)
  3. 计算库优化:cuBLAS/cuDNN等加速库针对N卡有极致优化

可以把英伟达的驱动和CUDA的关系类比为Python解释器和Python语言。具体来说:
英伟达驱动(NVIDIA Driver):这个类似于Python解释器。它是底层的软件,负责与硬件(即GPU)进行通信。英伟达驱动提供了基本的功能和接口,使得操作系统和其他软件能够使用GPU进行各种计算任务。

CUDA(Compute Unified Device Architecture):这个类似于Python语言。CUDA是一个并行计算平台和编程模型,它允许开发人员使用C、C++和Fortran等高级编程语言编写能够在英伟达GPU上运行的程序。CUDA依赖于英伟达驱动来访问和控制GPU硬件资源。


AMD显卡:ROCm

ROCm:AMD的开放计算平台(Radeon Open Compute)

对应组件 功能
ROCm Toolkit 包含编译器(HIPCC)、数学库(rocBLAS)等
MIOpen 深度学习加速库(对标cuDNN)
ROCm Driver AMD GPU内核级驱动

关键对比(NVIDIA CUDA vs AMD ROCm)

特性 NVIDIA CUDA AMD ROCm
启动时间 2006年 2016年
硬件兼容性 全系N卡 仅RX Vega及更新架构
容器支持 完善的NGC Docker镜像 有限支持
主流框架覆盖率 100% 约70%
典型延迟优化 纳秒级 微秒级

安装流程

1. 检查NVIDIA驱动及CUDA(NVIDIA显卡用户)

  • 安装专有NVIDIA驱动,之后重开机。
  • 安装 CUDA
  • 命令
    nvidia-smi       # 检查NVIDIA驱动状态
    nvcc --version   # 检查CUDA版本
    
  • 作用:确认GPU驱动和CUDA环境正常。
  • 问题处理
    • 若命令未找到:需安装NVIDIA驱动和CUDA。
    • 版本不兼容:确保CUDA版本与PyTorch兼容(如CUDA 11.x/12.x)。

2. 安装AMD驱动及ROCm(AMD显卡用户)

AMD显卡知识库
在 Ubuntu 系统上安装和配置 AMD GPU 驱动及 ROCm(Radeon Open Compute)平台


1. 安装内核头文件和额外模块
sudo apt install linux-headers-`uname -r	

相关文章:

  • Dify 是什么?Dify是一个开源的LLM应用开发平台,支持快速搭建生成式AI应用,具有RAG管道、Agent功能、模型集成等特点
  • 计算机网络,大白话
  • 代码随想录算法【Day44】
  • 2.13学习记录
  • Docker Desktop Windows 之 安装 SqlServer
  • RabbitMQ 延迟队列
  • 全功能Python测试框架:pytest
  • 使用 AlexNet 实现图片分类 | PyTorch 深度学习实战
  • 刷题日记---二叉树递归专题
  • OpenEuler学习笔记(三十):在OpenEuler上搭建3D建模环境
  • Linux内核实时机制x - 中断响应测试 Cyclictest分析1
  • 平面与平面相交算法杂谈
  • kafka的架构和工作原理
  • 在Mac arm架构终端中运行 corepack enable yarn 命令,安装yarn
  • 25、深度学习-自学之路-卷积神经网络基于MNIST数据集的程序展示
  • 14.1 Auto-GPT 项目定位与价值解读:揭开自主智能体的神秘面纱
  • 商品按图搜索拍立淘API接口系列概述
  • matplotlib中pyplot的基本使用
  • jvm 线程监控调试
  • Qt的isVisible ()函数介绍和判断窗口是否在当前界面显示
  • 网站建设同行友情链接/百度指数怎么下载
  • 吉林网站建设制作/网站关键词优化排名公司
  • 邪恶做动态网站/百度seo排名点击器app
  • 柏乡网站建设/seo网站优化建议
  • wordpress基础服务器/甘肃seo网站
  • 个体户备案网站可以做企业站吗/谷歌网址