讲师中心 微信公众号
AI工具推荐 视频效率加速

Google Colab怎么部署SD 谷歌Colab免费运行Stable Diffusion代码【教学】

陌晨吖_7031

陌晨吖_7031

发布时间:2025-12-16 15:43:02

|

906人浏览过

|

来源于php中文网

原创

可在Google Colab部署Stable Diffusion的五种方法:一、用diffusers+accelerate自动加载精简模型;二、挂载Drive手动加载本地权重;三、运行AUTOMATIC1111 WebUI封装版;四、启用xformers加速降显存;五、结合Drive实现模型与输出持久化。

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜

google colab怎么部署sd 谷歌colab免费运行stable diffusion代码【教学】

如果您希望在 Google Colab 上运行 Stable Diffusion 模型,但尚未配置环境或遇到运行失败、显存不足、依赖缺失等问题,则可能是由于运行时类型未正确设置、模型权重未自动下载、或代码未适配 Colab 的临时文件系统。以下是部署 Stable Diffusion 的多种可行方法:

一、使用 Hugging Face diffusers + accelerate 快速启动

该方法基于官方 diffusers 库,无需手动下载大模型文件,支持自动从 Hugging Face Hub 加载精简版 Stable Diffusion 模型(如 runwayml/stable-diffusion-v1-5),并利用 accelerate 进行显存优化。

1、在 Colab 新建 Notebook,点击「运行时」→「更改运行时类型」,将「硬件加速器」设为 GPU。

2、依次运行以下代码块:

!pip install --quiet diffusers transformers accelerate safetensors torch torchvision

3、执行模型加载与推理代码,注意指定 torch_dtype=torch.float16 以节省显存。

4、生成图像后,使用 from PIL import Image; image.save("output.png") 保存结果,并通过 files.download() 下载到本地。

二、挂载 Google Drive 手动加载本地模型权重

该方法适用于已下载完整模型权重(如 stable-diffusion-v1-5.ckpt 或.safetensors 文件)的用户,可绕过 Hub 下载限速与网络不稳定问题,同时复用已有的 LoRA、ControlNet 等扩展文件。

1、运行 from google.colab import drive; drive.mount('/content/drive') 挂载个人云盘。

2、将模型文件放入 Google Drive 根目录下的 /content/drive/MyDrive/sd_models/ 路径中。

3、在 Colab 中执行 !ln -s /content/drive/MyDrive/sd_models /content/sd_models 创建软链接,便于路径引用。

4、使用 diffusers.DiffusionPipeline.from_single_file 直接加载本地 .ckpt 或 .safetensors 文件,需指定 torch_dtype=torch.float16 和 use_safetensors=True(若为 safetensors 格式)。

三、使用 AUTOMATIC1111 WebUI 的 Colab 封装版本

该方法复现本地 WebUI 交互体验,支持文生图、图生图、Inpainting、模型切换、插件启用等功能,适合需要图形界面操作的用户,底层仍基于 diffusers 或原生 PyTorch 实现。

1、运行 !git clone https://github.com/camenduru/stable-diffusion-webui-colab 克隆社区维护的轻量封装仓库。

GitHub Actions SHA Rerun Debt Audit
GitHub Actions SHA Rerun Debt Audit

按提交SHA审查重跑成本,识别跨工作流反复消耗CI分钟数的提交。

下载

2、执行 !cd stable-diffusion-webui-colab && bash webui.sh --no-half 启动脚本,禁用半精度可避免部分 GPU 的兼容性报错。

3、等待日志中出现 Running on local URL: http://127.0.0.1:7860 后,点击 Colab 左侧「显示代理」按钮,选择对应端口链接进入 WebUI。

4、首次加载可能耗时较长,因需自动下载模型、VAE、Embeddings 等组件;后续运行将直接复用缓存。

四、启用 xformers 加速推理并降低显存占用

xformers 是由 Meta 开发的高效注意力库,可显著减少 Stable Diffusion 在 Colab T4/A100 GPU 上的显存峰值,并提升每秒生成步数(it/s),尤其适用于高分辨率输出或批量生成场景。

1、在安装依赖阶段追加命令:!pip install --quiet xformers。

2、加载 pipeline 后立即调用 pipe.enable_xformers_memory_efficient_attention() 方法(仅对支持的模型结构有效)。

3、若提示 xformers not available,需确认运行时为 GPU 类型且已重启运行时环境。

4、启用后,可通过 pipe.to("cuda") 显式迁移至 GPU,并观察日志中是否出现 Using xformers attention 提示。

五、处理 Colab 运行时中断与模型持久化问题

Colab 免费版运行时会在无活动 90 分钟后自动断开,且所有 /content/ 下文件在会话结束时被清除,导致模型反复下载、进度丢失。该方法通过组合 Google Drive 缓存与检查点保存机制保障连续性。

1、创建持久化目录:!mkdir -p /content/drive/MyDrive/sd_cache /content/drive/MyDrive/sd_outputs。

2、设置环境变量指向缓存路径:import os; os.environ['TRANSFORMERS_CACHE'] = '/content/drive/MyDrive/sd_cache'。

3、在生成循环中定期保存中间图像:if i % 5 == 0: image.save(f'/content/drive/MyDrive/sd_outputs/step_{i}.png')。

4、运行前校验模型是否存在:os.path.exists('/content/drive/MyDrive/sd_models/model.safetensors'),避免重复下载。

相关文章

谷歌浏览器
谷歌浏览器

谷歌浏览器Google Chrome是一款可让您更快速、轻松且安全地使用网络的浏览器。Google Chrome的设计超级简洁,使用起来得心应手。这里提供了谷歌浏览器纯净安装包,有需要的小伙伴快来保存下载体验吧!

下载

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

热门AI工具

更多
咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

Lovart
Lovart Hot

一款面向视觉设计创作的AI设计平台,可通过智能体和画布工作流辅助制作海报、Logo、网页、PPT及其他视觉内容。

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

相关专题

更多
PDF转图片方法
PDF转图片方法

需要把 PDF 页面用于上传、预览、分享或图片归档时,PDF 转图片方法专题整理 JPG/PNG 格式选择、逐页导出、清晰度设置、批量下载和结果检查等流程,帮助用户稳定完成 PDF 图片化处理。

0

2026.09.30

PixTV AI视频生成与无限画布创作
PixTV AI视频生成与无限画布创作

PixTV专题整理AI视频与视觉内容创作相关功能使用教程,涵盖AI生图、视频生成、无限画布、多模型创作、素材管理、声音音乐及视频剪辑等功能,帮助用户快速掌握PixTV从创意到成片的完整制作方法。

0

2026.09.29

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

200

2026.09.23

Buffalo框架路由与请求处理实操指南
Buffalo框架路由与请求处理实操指南

本专题讲解Buffalo框架路由与请求处理机制,涵盖路由注册与分组、资源路由、Handler编写规范、Context上下文方法、参数绑定、中间件编写挂载、Session与Cookie读写、Flash消息及错误页面定制方法。

120

2026.09.23

Buffalo框架零基础入门教程
Buffalo框架零基础入门教程

本专题整理Buffalo框架入门内容,涵盖Go环境准备、buffalo CLI安装、新项目生成、目录结构说明、dev热加载启动、数据库连接配置与常见报错排查,帮助新手按约定优于配置的思路跑通第一个Buffalo框架应用。

100

2026.09.23

Conan创建软件包配方指南
Conan创建软件包配方指南

本专题介绍通过conanfile.py创建软件包的方法,讲解包名、版本、依赖和构建设置等基础信息,以及source、build、package、package_info等常用方法的作用及编写思路。

60

2026.09.22

Conan二进制包配置指南
Conan二进制包配置指南

本专题介绍Conan根据操作系统、编译器、架构和构建类型生成二进制包的方法,讲解Profile、Settings、Options及Package ID的作用,帮助管理不同平台和编译环境下的包版本。

80

2026.09.22

Conan私有仓库搭建教程
Conan私有仓库搭建教程

本专题系统的讲解Conan私有仓库的搭建流程,涵盖仓库服务部署、存储目录配置、用户认证、权限划分和远程地址添加,并介绍内部C++依赖包的上传、下载及版本维护方法。

60

2026.09.22

loomy官网入口地址合集
loomy官网入口地址合集

本专题汇总了 Loomy 桌面 AI 助理的官方入口地址合集及使用指南。提供 macOS 与 Windows 客户端下载 。Loomy 是讯飞推出的桌面级 AI 工作搭子,支持文件整理、数据分析、网页操作及通过飞书/钉钉远程操控电脑,助你高效完成本地办公任务 。

80

2026.09.22

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
vscode手册
vscode手册

共0课时 | 0人学习

Git 教程
Git 教程

共21课时 | 7.9万人学习

Git版本控制工具
Git版本控制工具

共8课时 | 1.8万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn