0

0

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

幻夢星雲

幻夢星雲

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

|

826人浏览过

|

来源于php中文网

原创

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

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 免费无限量使用 DeepSeek R1 模型☜☜☜

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.float16use_safetensors=True(若为 safetensors 格式)。

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

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

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

MyMap AI
MyMap AI

使用AI将想法转化为图表

下载

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'),避免重复下载。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
go语言 注释编码
go语言 注释编码

本专题整合了go语言注释、注释规范等等内容,阅读专题下面的文章了解更多详细内容。

2

2026.01.31

go语言 math包
go语言 math包

本专题整合了go语言math包相关内容,阅读专题下面的文章了解更多详细内容。

1

2026.01.31

go语言输入函数
go语言输入函数

本专题整合了go语言输入相关教程内容,阅读专题下面的文章了解更多详细内容。

1

2026.01.31

golang 循环遍历
golang 循环遍历

本专题整合了golang循环遍历相关教程,阅读专题下面的文章了解更多详细内容。

0

2026.01.31

Golang人工智能合集
Golang人工智能合集

本专题整合了Golang人工智能相关内容,阅读专题下面的文章了解更多详细内容。

1

2026.01.31

2026赚钱平台入口大全
2026赚钱平台入口大全

2026年最新赚钱平台入口汇总,涵盖任务众包、内容创作、电商运营、技能变现等多类正规渠道,助你轻松开启副业增收之路。阅读专题下面的文章了解更多详细内容。

76

2026.01.31

高干文在线阅读网站大全
高干文在线阅读网站大全

汇集热门1v1高干文免费阅读资源,涵盖都市言情、京味大院、军旅高干等经典题材,情节紧凑、人物鲜明。阅读专题下面的文章了解更多详细内容。

73

2026.01.31

无需付费的漫画app大全
无需付费的漫画app大全

想找真正免费又无套路的漫画App?本合集精选多款永久免费、资源丰富、无广告干扰的优质漫画应用,涵盖国漫、日漫、韩漫及经典老番,满足各类阅读需求。阅读专题下面的文章了解更多详细内容。

67

2026.01.31

漫画免费在线观看地址大全
漫画免费在线观看地址大全

想找免费又资源丰富的漫画网站?本合集精选2025-2026年热门平台,涵盖国漫、日漫、韩漫等多类型作品,支持高清流畅阅读与离线缓存。阅读专题下面的文章了解更多详细内容。

19

2026.01.31

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Git 教程
Git 教程

共21课时 | 3.2万人学习

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

共8课时 | 1.5万人学习

Git中文开发手册
Git中文开发手册

共0课时 | 0人学习

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号