之前曾介紹過在Ubuntu22.04上安裝Stable-Diffusion:
在Ubuntu22.04上部署Stable Diffusion_ubuntu stable dif-CSDN博客
這個安裝我們使用conda python虛擬機。這次我們介紹的是在Ubuntu24.04安裝Stable-Diffusion的最新版本V1.10.1(截止到今天最新版),并且我們這次安裝不再使用conda虛擬環境。
一:安裝GPU
1、安裝編譯環境:
apt update
apt install gcc g++
apt install make
2、禁用nouveau
vim /etc/modprobe.d/blacklist.conf#在文件的最后添加下面兩行。blacklist nouveau
options nouveau modeset=0#然后,輸入下面的命令更新并重啟。update-initramfs -u
reboot#重啟后輸入下面的命令驗證是否禁用成功,成功的話這行命令不會有輸出。lsmod | grep nouveau#驅動安裝
#首先,使用apt卸載已有的驅動,命令如下。apt-get purge nvidia*#進入驅動所在路徑,賦予執行權限,并執行安裝命令chmod +x NVIDIA-Linux-x86_64-535.86.05.run
./NVIDIA-Linux-x86_64-535.86.05.run
3、安裝cuda
#下載cuda
wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run#為安裝文件附權限
chmod +x cuda_12.4.0_550.54.14_linux.run#運行安裝文件
./cuda_12.4.0_550.54.14_linux.run
配置環境變量
#重啟服務器后,如不生效,可直接將文件寫在.bashrc里面
vim ~/.bashrc#添加以下幾句:PATH=$PATH:/usr/local/cuda/bin
LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64#使生效
source ~/.bashrc
4、安裝cudnn
#解壓文件
xz -d cudnn-linux-x86_64-8.9.7.29_cuda12-archive.tar.xz
tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda12-archive.tar#拷貝文件
cp /root/cudnn-linux-x86_64-8.9.7.29_cuda12-archive/include/cudnn.* /usr/local/cuda/include/
cp /root/cudnn-linux-x86_64-8.9.7.29_cuda12-archive/lib/* /usr/local/cuda/lib64/
chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
相關文件的下載路徑,請參考之前的相關文檔。?
二:安裝python3.10
1、安裝python的相關依賴
apt update
apt install build-essential zlib1g-dev libncurses5-dev libgdbm-dev libnss3-dev libssl-dev libreadline-dev libffi-dev libsqlite3-dev wget libbz2-dev
2、從官網下載python3.10
wget https://www.python.org/ftp/python/3.10.0/Python-3.10.0.tgz
3、解壓文件編譯
#解壓文件
tar -zvxf Python-3.10.0.tgz#安裝編譯工具
apt install gcc g++
apt install make#編譯代碼
cd Python-3.10.0/
./configure --enable-optimizations
make#開始安裝
make install
4、查看python版本
python3.10 --version
查看python的安裝位置
which python3.10
?
python3.10的安裝路徑:?/usr/local/bin/python3.10
pip3.10的安裝路徑:/usr/local/bin/pip3.10
5、設置軟連接
ln -s /usr/local/bin/python3.10 /usr/bin/python
ln -s /usr/local/bin/pip3.10 /usr/bin/pip
6、設置python環境
vim ~/.bashrc#添加:
alias python='/usr/local/bin/python3.10'
使配置生效:
#使配置生效:
source ~/.bashrc#運行查看:
python#退出
>>>exit();
三:安裝Stable-Diffusion
1、下載stable-diffusion代碼庫
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui && cd stable-diffusion-webui
2、安裝相關依賴
pip install -r requment.txt
3、啟動Stable-Diffusion
cd /app/stable-diffusion-webui/
python launch.py
四:設置開機啟動
vim /etc/systemd/system/stable-diffusion.service
添加以下內容?
[Unit]
Description=Run Python script main.py
After=network.target[Service]
ExecStart=/usr/bin/python /app/stable-diffusion-webui/launch.py
Restart=always
User=root[Install]
WantedBy=multi-user.target
注意python的路徑和launch.py的路徑。
systemctl daemon-reload
systemctl start stable-diffusion.service
systemctl enable stable-diffusion.service
五:報錯處理
ModuleNotFoundError: No module named '_lzma'
安裝torch==1.9.0和torchvision==0.10.0時會報錯 ModuleNotFoundError: No module named '_lzma',處理方法:
apt install liblzma-dev -y
pip install backports.lzma
編輯lzma.py
#修改前
from _lzma import *
from _lzma import _encode_filter_properties, _decode_filter_properties#修改后
try:from _lzma import *from _lzma import _encode_filter_properties, _decode_filter_properties
except ImportError:from backports.lzma import *from backports.lzma import _encode_filter_properties, _decode_filter_properties
注意行的縮進
保存后,重啟服務器,然后再執行python launch.py
http://IP:7860