一、事前準備
1、一臺配置不錯的電腦,英偉達顯卡,20系列起步,建議顯存6G起步,安裝win10或以上版本,我的顯卡是40系列,16G顯存,所以跑大部分的模型都比較快;
2、科學上網,很多下載所需的鏈接都在國外,雖然國內目前資源也不少了,但是精品不多,建議備一個。
二、安裝所需的環境
具體包括python,這是Stable Diffusion的主要運行環境;git,用來克隆下載所需項目,包括Stable Diffusion自身;cuda,英偉達顯卡用來跑算法的驅動,沒有這個,顯卡就是擺設。
1、安裝python
記住,必須是3.10.6版本,后期的版本不支持pytorch,無法跑顯卡算法,部署上會一直報錯。
下載鏈接:Python Release Python 3.10.6 | Python.org
下載后按提示一路安裝,安裝完把安裝路徑加入到系統變量中,按如下操作步驟:
在cmd命令行中敲python --version,顯示如下結果即說明安裝好了。
2、安裝git
到官網Git?下載軟件,按提示安裝。
如何使用git參考我之前的一篇文章:手記:把代碼上傳到Gitee等遠程倉庫的過程記錄及常見問題_提交到gitee倉庫的代碼看不到嗎-CSDN博客
安裝完敲入指令git --version看版本
3、安裝cuda
到官網CUDA Toolkit Archive | NVIDIA Developer?下載驅動,按提示安裝。這里需要為自己的顯卡選擇具體的版本,命令行輸入指令nvidia-smi查看對應版本。
下載后安裝完也要配置一下環境變量,前面配置python環境變量的截圖中有,操作步驟雷同。
三、安裝與配置Stable Diffusion
1、下載Stable Diffusion
使用cmd,進入想安裝的目錄,敲入指令:
git clone https://github.com/Stability-AI/stablediffusion.git
2、下載所需的包,配置運行環境
運行指令:
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple ?
查看這個txt文檔,內容如下
最好是手動一條條安裝,如pip install?albumentations==0.4.3
這樣好能看出哪個安裝有問題,以便定向排查問題。
3、下載基礎模型
目錄如下:
新建目錄models,在其中再建個Stable-diffusion目錄,用于存放基礎模型文件。
到網站?stable-diffusion-v1-5/stable-diffusion-v1-5 at main上下載個基礎模型。
?
下載的模型文件放入models\Stable-diffusion目錄中
4、運行畫圖指令
把ldm目錄移入scripts目錄中,不然運行下面的指令時會報no ldm modules的錯誤。
python ./scripts/txt2img.py --prompt "a man" --ckpt ./models/Stable-diffusion/v1-5-pruned.ckpt --outdir outputs --H 512 --W 512
運行完圖片生成在outputs中。
?
至此,專業版的部署已經完成,雖然這種方式有利于理解它背后的運行機制,用于研究。不過使用的時候每次都要編寫指令,在cmd窗口中運行,或者使用python代碼從程序中繪畫,很不便捷,下一篇介紹如何使用web頁面畫圖。