ChilloutMix騰訊云服務(wù)器快速部署

KyunraWang 2023-03-14 發(fā)布于四川

展開全文

背景

前兩天在某公眾號的軟文刷到了AI繪畫領(lǐng)域最新的ChilloutMix模型。大概是下面這張圖的效果：

激動的心顫抖的手，學(xué)了這么多年計算機(jī)，就是為了走在時代前沿，事到如今終于可以活學(xué)活用了

然而自己的電腦配置完全不夠用，咋整？開臺GPU云服務(wù)器吧。

踩了無數(shù)坑之后，終于總結(jié)出此流程。力求以最簡單的方式，快捷的在云服務(wù)器部署chilloutmix，以及l(fā)ora，以及各種其他繪畫模型。

在騰訊云實測，從創(chuàng)建機(jī)器到完成部署，總時長在30分鐘左右。

廢話不多說，下面進(jìn)入正題

一. 購買服務(wù)器（3-5分鐘）

首先需要購買一臺騰訊云的GPU服務(wù)器。GPU卡型本次選擇“T4”，性能完全夠用，對應(yīng)“GN7”機(jī)型。我是從GPU實驗室活動頁搶購到的折扣機(jī)型（GPU實驗室:https://cloud.tencent.com/act/pro/gpu-study?from=10680）

注意，點擊立即購買后，預(yù)裝鏡像要選擇“Ubuntu Server 18.04.1”，之后系統(tǒng)會默認(rèn)自動安裝GPU驅(qū)動。若選擇其他鏡像，可能會導(dǎo)致無法順利跑通本流程。

（如果子彈充足，或者做圖心切，也可直接從騰訊云官網(wǎng)購買GPU服務(wù)器(https://cloud.tencent.com/product/gpu)，豐儉由人。建議選按量付費機(jī)型即可，鏡像選擇Ubuntu20.04，勾選“后臺自動安裝GPU驅(qū)動”，驅(qū)動版本等按默認(rèn)配置。）

二. 前置配置（10-15分鐘）

1.登錄機(jī)器后，會默認(rèn)進(jìn)行驅(qū)動的自動安裝，可以用以下命令查看當(dāng)前安裝進(jìn)程，該步驟需要5～10分鐘：

ps aux | grep -i install

自動安裝順序為（1）GPU驅(qū)動；（2）CUDA；（3）cuDNN

2.當(dāng)GPU驅(qū)動、CUDA、cuDNN都安裝完成后，通過以下命令創(chuàng)建python3虛擬環(huán)境，完成前置配置

sudo apt install wget git python3 python3-venv

三. 部署webui，完成配置（10-15分鐘）

1.將webui clone到服務(wù)器本地：

git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui

2.成功后，進(jìn)入stable-diffusion-webui文件夾：

cd stable-diffusion-webui

3.在launch.py的代碼中，由于騰訊源沒有“torch==1.13.1+cu117“，導(dǎo)致需要從“--extra-index-url”的“https://download.pytorch.org/whl/cu117”中進(jìn)行下載，速度非常慢。故此處對launch.py代碼進(jìn)行修改：

原代碼：torch_command = os.environ.get('TORCH_COMMAND', "pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117")

修改后代碼：torch_command = os.environ.get('TORCH_COMMAND', "pip install torch==1.13.1 torchvision==0.14.1")

對不熟悉linux操作的朋友們，可參考如下手把手命令行教學(xué)：

vim launch.py

輸入"i"找到這行：torch_command = os.environ.get('TORCH_COMMAND', "pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117")，并修改

按"esc"輸入":wq"

完成

4.下載chilloutmix模型：進(jìn)入stable-diffusion-webui/models/Stable-diffusion文件夾，通過如下命令行下載chilloutmix模型到本地。該步驟下載速度較快，大約需要3-5min即可完成

cd stable-diffusion-webui/models/Stable-diffusion wget -O chilloutmix_NiPrunedFp32.safetensors https:///api/download/models/9475

5.最后，進(jìn)入stable-diffusion-webui文件夾，輸入激動人心的“./webui.sh”。通過自動化腳本完成安裝。該步驟會自動安裝webui所需的各依賴項，如gfpgan，k_diffusion等。實測完成全部安裝大概需要10-20min。

./webui.sh

若希望生成可以分享給他人使用的的鏈接，在./webui.sh后增加 "--share"即可

./webui.sh --share

如果順利的話，等待10分鐘左右，全部安裝完成后就可以看到鏈接了

提示: 由于服務(wù)器和github連接不是很穩(wěn)定，中途可能出現(xiàn)網(wǎng)絡(luò)錯誤。若出現(xiàn)網(wǎng)絡(luò)問題，重新輸入“./webui.sh”即可。或可通過報錯提示信息，手動執(zhí)行相應(yīng)步驟安裝后，重新輸入“./webui.sh”繼續(xù)剩余步驟的自動安裝。

此處若有問題，可在評論區(qū)評論，看到后都會回復(fù)～