簡介
由於 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 ... 11-linux_x86_64.whl
https://download.pytorch.org/whl ... 11-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 [url]https://download.pytorch.org/whl/rocm6.0'[/url] python3 launch.py --precision full --no-half --skip-torch-cuda-test
# stable-diffusion-webui 官方方法,有缺陷
#TORCH_COMMAND='pip install torch torchvision --extra-index-url [url]https://download.pytorch.org/whl/rocm6.0'[/url] 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"