最近AI繪畫很火，所以想在本地部署個環(huán)境體驗一下。原本打算在Windows下安裝，網(wǎng)上有很多教程，也有很多大佬做了整合包，可是我的Windows系統(tǒng)下的顯卡驅(qū)動版本太低，也沒法升級，只能裝CUDA9，導(dǎo)致安裝torch會出現(xiàn)問題。所以考慮在Ubuntu下安裝，參考了幾篇文章，所幸也安裝成功了。這里記錄一下安裝過程，以供參考。

我的機(jī)器配置：

?1.安裝cuda

支持安裝的cuda版本可以用nvidia-smi命令查看

我的顯卡驅(qū)動支持的cuda版本最高是11.4，可以在這里下載對應(yīng)的cuda版本：CUDA Toolkit Archive | NVIDIA Developer。我由于之前安裝別的環(huán)境，已經(jīng)裝了cuda10.2版本，所以我就直接用這個版本了。可以用nvcc -V查看安裝的cuda版本。?

2.搭建python虛擬環(huán)境

使用miniconda，我之前已經(jīng)安裝了。然后使用conda創(chuàng)建虛擬環(huán)境sdwebui，使用的python版本為3.10.9（stable-diffusion-webui用的python版本為3.10.6）。

conda create -n sdwebui python==3.10.9

激活環(huán)境

conda activate sdwebui

升級pip并更改默認(rèn)庫包下載地址為清華鏡像

python -m pip install --upgrade pip
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

安裝torch

在pytorch官網(wǎng)Previous PyTorch Versions | PyTorch

查看對應(yīng)的cuda支持的torch版本，并且直接拷貝安裝命令。

這里是我的cuda版本對應(yīng)的安裝命令。?

3.下載并安裝stable-diffusion-webui

切換到當(dāng)前用戶目錄下，使用git命令下載stable-diffusion-webui源碼

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

由于國內(nèi)github不能穩(wěn)定連接，所以可以從國內(nèi)的gitee（碼云）下載

git clone https://gitee.com/yiouyou/automatic1111-stable-diffusion-webui.git

打開v2-1_768-ema-pruned.ckpt · stabilityai/stable-diffusion-2-1 at main，下載訓(xùn)練模型。下載完成后，把v2-1_768-ema-pruned.ckpt這個訓(xùn)練模型放入stable-diffusion-webui的models/Stable-diffusion目錄下，這個目錄專門存放用于生成AI繪圖的繪圖元素的基礎(chǔ)模型庫。后續(xù)如果在其他網(wǎng)站比如civitai之類的地方下載的ckpt或者safetensors文件也是放在這個文件夾里面。

然后切換到webui目錄下，安裝依賴

cd stable-diffusion-webui/
pip install -r requirements_versions.txt
pip install -r requirements.txt

安裝完成后，啟動stable-diffusion-webui

python launch.py

這里要安裝很多依賴，并且要git clone相關(guān)包和模型文件，需要科學(xué)上網(wǎng)，否則會失敗。

最終啟動成功后如下圖所示，會出來一個url

?在瀏覽器打開這個url，如圖所示，表示環(huán)境部署成功

?4.使用stable-diffusion-webui

在左上角的文本框prompt輸入提示詞，點擊Generate，等待數(shù)秒鐘（我的電腦大概等待15s左右），就會生成一張圖。

注意：如果在prompt輸入后點擊generate無法生成圖片，看見黑色小框顯示“float 32“之類的關(guān)鍵字，則在webui頁面的Settings->StableDiffusion最下面，勾選float32的選項框。接著點擊上方的Apply settings應(yīng)用，然后按F5刷新頁面后，即可正常使用。

目前只是初步體驗了一下，下一步會繼續(xù)學(xué)習(xí)，繼續(xù)解鎖新玩法。

參考：

Ubuntu安裝stable-diffusion-webui詳細(xì)教程 - 知乎

從零開始，手把手教你本地部署Stable Diffusion Webui AI繪畫(Win系最新版) - 知乎

其他：

AI繪畫咒語tag在線生成器：自己去找對應(yīng)的tag生成prompt

https://civitai.com：這個里面有很多好看的模型文章來源地址http://www.zghlxwxcb.cn/news/detail-410127.html

到了這里，關(guān)于Ubuntu20.04本地部署Stable-Diffusion-webui AI繪畫環(huán)境的文章就介紹完了。如果您還想了解更多內(nèi)容，請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！