惯性聚合 高效追踪和阅读你感兴趣的博客、新闻、科技资讯
阅读原文 在惯性聚合中打开

推荐订阅源

Engineering at Meta
Engineering at Meta
J
Java Code Geeks
I
InfoQ
腾讯CDC
Vercel News
Vercel News
IT之家
IT之家
V
Visual Studio Blog
P
Proofpoint News Feed
freeCodeCamp Programming Tutorials: Python, JavaScript, Git & More
G
Google Developers Blog
Cyber Security Advisories - MS-ISAC
Cyber Security Advisories - MS-ISAC
博客园 - 叶小钗
有赞技术团队
有赞技术团队
月光博客
月光博客
Martin Fowler
Martin Fowler
量子位
L
LangChain Blog
B
Blog
Last Week in AI
Last Week in AI
博客园 - 司徒正美
Microsoft Security Blog
Microsoft Security Blog
博客园 - 聂微东
Microsoft Azure Blog
Microsoft Azure Blog
A
About on SuperTechFans

博客园 - opama

AI智能用例整理 AI 执行测试用例 AI 编写接口自动化用例 基于知识库的需求评审到用例生成的智能辅助 AI 在工作中的一些使用 不要忘记好的测试也是个业务专家 打造个人面试助手 看数类需求测试方法 AI辅助测试用例生成 利用GPT快速生成自动化代码 本地部署AI知识库 基于自己知识库的机器人试用 线上问题SLA治理 AI绘图新手入坑 airtest使用案例一则 airtest初步使用 测试设计左移实践 接口越权扫描平台初探 通过阅读代码来优化测试执行 左移白盒测试实践 通过机器人通知提效的那些事情 《敏捷革命》读书感想 《SAAS创业指南》拆书笔记——产品打磨和商业模式初步验证
A卡在ubuntu下部署stable-diffusion-webui
opama · 2023-06-04 · via 博客园 - opama

因为自己之前为了装黑苹果把1080ti卖了买了6800XT,在现在这个玩AI的时代后悔莫及,先尝试在mac m1下安装了stable-diffusion-webui,功能基本上都能用,就是速度太慢。
后来想了想还是装了ubuntu,组成win+mac+ubuntu的三系统

1. 安装ubuntu

安装ubuntu基本都有教程,使用UEFI安装好之后在启动器上配置下ubuntu的启动位置,不然启动系统后看不到ubuntu系统

2. 安装显卡驱动和rocm

安装stable-diffusion-webui之前需要安装AMD的驱动和rocm,主要参考:

安装过程有点慢,我晚上挂机安装

3. 安装pyTorch

https://pytorch.org/get-started/locally/ 安装pyTorch,选择rocm的方式

安装完检查下安装是否正常
进入python

python3

在python中输入,

import torch
torch.cuda.is_available() 

如果返回True,说明GPU有官方支持,直接exit()退出python

4. 安装stable-diffusion-webui

参考官方文档或者https://www.bilibili.com/read/cv22056432 进行安装
然后正常安装stable-diffusion-webui

需要注意
打开webui-user.sh文件进行编辑,找到export COMMANDLINE_ARGS这一行,在后面写上"--precision full --no-half --skip-torch-cuda-test"
这个不同显卡型号不一样,6800XT是可以不加--precision full --no-half ,其他型号可以先加上正常启动后然后去掉看下

启动时遇到:

MIOpen(HIP): Warning [BuildHip] /tmp/comgr-5b498c/input/naive_conv.cpp:39:10: fatal error: 'limits' file not found
#include <limits> // std::numeric_limits
         ^~~~~~~~
1 error generated when compiling for gfx1030.
terminate called after throwing an instance of 'miopen::Exception'

可以参考
https://github.com/RadeonOpenCompute/ROCm/issues/1889
sudo apt install libstdc++-12-dev

5.运行时检查是否用的是GPU

rocm-smi看下是不是使用的是GPU,第一次装没有装好rocm直接用的是CPU,巨慢无比

======================= ROCm System Management Interface =======================
================================= Concise Info =================================
GPU  Temp (DieEdge)  AvgPwr  SCLK     MCLK     Fan     Perf  PwrCap  VRAM%  GPU%  
0    66.0c           269.0W  2325Mhz  1000Mhz  54.12%  auto  255.0W   71%   99%   

实测下来,512x512 放大2倍,mac需要10min,而6800XT只需要1min,速度提升很大,比较满意