OS Ubuntu 22.04.4 LTS 64-bit
GPU AMD® Radeon rx 7600
前言
Ubuntu 22.04 安装 ROCm6.0教程+踩坑http://t.csdnimg.cn/d9vLb
由于之前已经安装了ROCm6.0, 所以后面部署SD用的都是6.0的依赖,也有些问题没有解决,如开启xFormers,主要是需要
PyTorch 2.2.2+cu121 with CUDA 1201 (you have 2.4.0.dev20240424+rocm6.0)
根据MarKA的文章,似乎5.7稳定支持,所以bug可能少一点?ROCm5.7请参考原文。
本文代码的步骤大量搬运了MarKA的文章,并根据遇到的问题做了修改和补充。
一、前期准备
1. 检查ROCm信息
- 显卡正常显示即表示驱动安装正常
rocminfo
2. 安装conda
- 任意conda版本皆可,本文使用MarKA的文章提到的miniconda
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash ./Miniconda3-latest-Linux-x86_64.sh
conda config --set auto_activate_base false #禁用自动进入conda环境
3. 创建conda环境
- 创建一个名为sd的环境,python版本为3.10.6(版本可自行指定):
conda create -n sd python=3.10.6 -y
cond