简介
由于 AMD 官方没有提供 Debian12 的驱动和 ROCM,只好安装 Ubuntu20.04 的驱动和 ROCM,必要软件 git 和 python3-venv。
添加 i386 仓库
sudo dpkg --add-architecture i386 && \
sudo apt upgrade -y && \
apt upgrade -y
下载驱动安装程序
到 AMD 官网下载 Ubuntu20.04 驱动安装程序:
https://www.amd.com/zh-hans/support/linux-drivers
- 一定要下载 Ubuntu20.04 驱动安装程序
安装驱动安装程序
- 到驱动目录后安装驱动安装程序
sudo apt install amdgpu-install_6.1.60103-1_all-u20.04.deb
安装驱动
sudo amdgpu-install --usecase=dkms,opencl,hip,rocm
- dkms:内核显卡驱动
- opencl,hip:显卡 hip 计算,例如 Blender 会用到
- rocm:rocm
- 安装后会有以下错误,不用关心,不影响使用
下列软件包有未满足的依赖关系:
rocm-gdb : 依赖: libpython3.10 但无法安装它 或
libpython3.8 但无法安装它
安装 AMD 显卡监控工具
sudo apt install radeontop rocm-smi
- 推荐使用 radeontop 信息比较全
使用 pip 国内镜像
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
安装 git 和 python3-venv
sudo apt install git python3-venv
下载 pytorch rocm 版
可以到:
https://download.pytorch.org/whl/torch/
https://download.pytorch.org/whl/pytorch-triton-rocm/
下载:
https://download.pytorch.org/whl/pytorch_triton_rocm-2.3.1-cp311-cp311-linux_x86_64.whl
https://download.pytorch.org/whl/rocm6.0/torch-2.3.1%2Brocm6.0-cp311-cp311-linux_x86_64.whl
使用 aria2c -x 16 可以加快点速度下载
创建 python3-venv 虚拟环境
mkdir $HOME/sd-rocm && python3 -m venv $HOME/sd-rocm
安装 stable-diffusion-webui
# 激活 python3-venv 虚拟环境
source $HOME/sd-rocm/bin/activate
# 安装 刚才下载的 pytorch,注意安装顺序
pip install pytorch_triton_rocm-2.3.1-cp311-cp311-linux_x86_64.whl
pip install torch-2.3.1+rocm6.0-cp311-cp311-linux_x86_64.whl
# 下载 stable-diffusion-webui
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui
# 进入 stable-diffusion-webui 目录
cd stable-diffusion-webui
# 第一次运行
TORCH_COMMAND='pip install torchvision --extra-index-url https://download.pytorch.org/whl/rocm6.0' python3 launch.py --precision full --no-half --skip-torch-cuda-test
# stable-diffusion-webui 官方方法,有缺陷
#TORCH_COMMAND='pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm6.0' python launch.py --precision full --no-half
# 以后直接运行,先激活虚拟环境在进入 stable-diffusion-webui 目录
python3 webui.py --precision full --no-half --skip-torch-cuda-test
错误处理
出现错误:
AttributeError: 'NoneType' object has no attribute 'lowvram'
这个缺少 clip-vit-large-patch14 有关,到 https://www.modelscope.cn 搜索 clip-vit-large-patch14,进入 stable-diffusion-webui 目录 创建 openai,进取 openai,git clone 刚才搜索到的 clip-vit-large-patch14 仓库网址即可。
缺少模型
到 https://www.modelscope.cn 搜索 v1-5-pruned-emaonly.safetensors,然后下载 v1-5-pruned-emaonly.safetensors 放到 stable-diffusion-webui/models/Stable-diffusion 目录下即可。
END
# FROM 小熊宝宝
echo "https://my.oschina.net/hemiya"
echo "https://www.cnblogs.com/cnhemiya"
标签:diffusion,--,AMD,rocm,webui,https,stable
From: https://www.cnblogs.com/cnhemiya/p/18314416