当前位置: 首页 > 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 DriverAMD GPU内核级驱动

关键对比(NVIDIA CUDA vs AMD ROCm)

特性NVIDIA CUDAAMD 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	

http://www.mrgr.cn/news/90689.html

相关文章:

  • Java的synchronized是怎么实现的?
  • 深入解析C++命名空间:从基础到高级应用
  • docker compose快速部署kafka-connect集群
  • 在 rtthread中,rt_list_entry (rt_container_of) 已知结构体成员的地址,反推出结构体的首地址
  • 【Vue中BUG解决】npm error path git
  • 解锁大语言模型潜能:KITE 提示词框架全解析
  • Ubuntu启动geteck/jetlinks实战:Docker启动
  • Proxmox VE 8.3 qm 方式导入ESXi Linux OVA UEFI模式虚拟机
  • C++类和对象进阶:运算符重载深度详解
  • 在vscode中拉取gitee里的项目并运行
  • Python----PyQt开发(PyQt高级:手搓一个文件浏览器)
  • Druid GetConnectionTimeoutException解决方案之一
  • vue-model如何自定义指令,及批量注册自定义指令
  • 【GRPO】GRPO原理原文翻译
  • docker学习---第3步:docker实操大模型
  • Hive增量迁移方案与实操PB级
  • Linux初始化 配置yum源
  • 大数据学习之PB级百战出行网约车二
  • poi 将图片写入到excel文件中
  • vue开发06:前端通过webpack配置代理处理跨域问题
  • NumPy中生成和堆叠数组、生成切片的特殊对象:np.r_ np.c_ np.s_
  • 欧盟《人工智能法案》
  • SolidWorks C# How
  • 大模型WebUI:Gradio全解12——使用Agents和Tools智能代理工具构建UI(1)
  • 系统URL整合系列【不改hosts文件版】-- 代码1(springcloud-gateway动态路由)
  • 教程 | Proxmox VE(PVE)安装全流程指南(末尾附镜像及快速配置脚本)