GPU LAB 煉丹秘籍｜快速部署 AI 繪畫

AI 繪畫模型（以 StableDiffusion 為首）自去年推出後快速迭代。近期，市面上又出現了一批效果驚豔的新突破。

目前，若希望運行 AI 繪畫模型，主要有以下幾種方式：

方式	優勢	劣勢
使用個人電腦或免費在線平臺運行	成本低	出圖效率低
使用付費在線平臺	出圖效率較高	成本高
使用雲服務器部署	出圖效率最高，且成本較低	部署有一定門檻

可以看出，使用雲服務器部署 AI 繪畫模型是綜合性價比最高的一種方式，但由於其使用的固有門檻，導致大多數煉丹師望而卻步。

本文將演示如何使用 GPU LAB，極簡部署 AI 繪畫模型，降低用戶在雲服務器部署 AI 繪畫模型的使用門檻。

購買服務器（3-5分鐘）

首先，需購買一臺騰訊雲的 GPU 服務器。GPU 卡型本次選擇 T4，對應“GN7”機型。本次從 GPU 實驗室活動頁購買特價的折扣機型（GPU實驗室）。

注意，購買時預裝鏡像選擇 Ubuntu Server 18.04.1，之後系統會默認自動安裝 GPU 驅動。

前置驅動自動安裝（10分鐘）

登錄機器後，大約需要10分鐘進行驅動的自動安裝，可以用以下命令查看當前安裝進程：

ps aux | grep -i install

若“當前進程”如下圖所示，表示所有前置安裝已完成。可進入下一步，開始 AI 繪畫環境部署。

開始部署

Plan A（推薦）：三條指令部署 AI 繪畫環境

在確保已完成驅動自動安裝後，可通過執行下述三條指令，快速部署 AI 繪畫環境：

首先，執行以下命令行，部署 docker 及 nvidia docker 環境：

(while true ; do ! (ps -aux | grep -i install | grep -v grep > /dev/null) && break; echo "Waitting for driver, cuda or cudnn install finished."; sleep 10; done;) &&(curl -fsSL http://mirrors.tencent.com/docker-ce/linux/ubuntu/gpg | sudo apt-key add -) && sudo add-apt-repository "deb [arch=amd64] http://mirrors.tencent.com/docker-ce/linux/ubuntu $(lsb_release -cs) stable" && sudo apt -y update && sudo apt-get install -y docker-ce  && sudo systemctl start docker && sudo systemctl enable docker && (curl -s -L http://mirrors.tencent.com/install/GPU/taco/get-nvidia-docker2.sh | sudo bash)

之後，執行以下命令行，拉取容器鏡像。容器鏡像中已完成 AI 繪畫所需環境配置，並預裝了 stable-diffusion-v1-5 模型文件（該步驟所需時間較久，約10-15min）。

sudo docker pull gpulab.tencentcloudcr.com/ai/stable-diffusion:1.0.3

最後，執行以下命令行啟動容器並啟動 AI 繪畫。將該步驟最後生成的鏈接粘貼至瀏覽器，即可開始使用。

sudo docker run -itd --gpus=all --network=host --device=/dev/dri --group-add=video --ipc=host --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --name=stable-diffusion gpulab.tencentcloudcr.com/ai/stable-diffusion:1.0.3 | xargs sudo docker logs --follow

注：若希望修改模型文件，可參考文末“其他”部分操作指引。

PlanB：一鍵完成 AI 繪畫環境部署

同 planA，在確保已完成驅動自動安裝後，可將 planA 中三條指令合為一條，一鍵完成 webui 部署。將該腳本執行最後生成的鏈接粘貼至瀏覽器，即可開始使用 AI 繪畫。

(while true ; do ! (ps -aux | grep -i install | grep -v grep > /dev/null) && break; echo "Waitting for driver, cuda or cudnn install finished."; sleep 10; done;) && (curl -fsSL http://mirrors.tencent.com/docker-ce/linux/ubuntu/gpg | sudo apt-key add -) && sudo add-apt-repository "deb [arch=amd64] http://mirrors.tencent.com/docker-ce/linux/ubuntu $(lsb_release -cs) stable" && sudo apt -y update && sudo apt-get install -y docker-ce  && sudo systemctl start docker && sudo systemctl enable docker && (curl -s -L http://mirrors.tencent.com/install/GPU/taco/get-nvidia-docker2.sh | sudo bash) && sudo docker pull gpulab.tencentcloudcr.com/ai/stable-diffusion:1.0.3 && sudo docker run -itd --gpus=all --network=host --device=/dev/dri --group-add=video --ipc=host --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --name=stable-diffusion gpulab.tencentcloudcr.com/ai/stable-diffusion:1.0.3 | xargs sudo docker logs --follow

使用方式

輸入“Prompt”，按個人喜好調整“sampling method”，“Sampling steps”等參數後，點擊“Generate”即可生成圖片。

其他

若希望使用其他模型：可在進入容器後，進入 stable-diffusion-webui/models/Stable-diffusion 文件夾，通過如下命令行下載模型到本地。模型文件可訪問如 https://huggingface.co/CompVis 或 https://civitai.com 等進行下載。

下載完成後，需退出容器並重啟，以重新啟動 AI 繪畫環境。

進入文件夾：cd stable-diffusion-webui/models/Stable-diffusion
下載指令：wget -O 模型名稱 模型下載鏈接
（例：wget -O SD-v6.ckpt https://civitai.com/api/download/models/xxxx）

容器相關操作指引：

進入容器

sudo docker exec -it stable-diffusion bash

退出容器

exit

重新生成 AI 繪畫環境分享鏈接

sudo docker restart stable-diffusion | xargs sudo docker logs --follow

本文中涉及的框架及軟件，由權利人根據其配套的開源許可證條款對外授權，在下載和使用之前應自行閱讀、理解並遵守適用的開源許可證條款，自行承擔違反該等開源許可證條款造成的所有責任和損失，並應保證使騰訊雲免於遭受由於您違反適用開源許可證條款所產生的責任和損失。