AI绘图@一键安装

这个脚本能够实现Linux系统中N卡环境搭建和table-diffusion-webui项目安装。

执行脚本

#!/bin/bash

# 检查是否已经安装了CUDA
if ! command -v nvcc &> /dev/null
then
    # 下载CUDA安装包
    wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.57.02_linux.run
    
    # 安装CUDA
    sudo sh cuda_11.4.0_470.57.02_linux.run --silent --toolkit
    
    # 将CUDA路径添加到环境变量中
    echo 'export PATH=/usr/local/cuda-11.4/bin:$PATH' >> ~/.bashrc
    echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
    source ~/.bashrc
fi

# 检查是否已经安装了cuDNN
if [ ! -d "/usr/local/cuda-11.4/include/cudnn.h" ]
then
    # 下载cuDNN安装包
    wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.57.02_linux.run
    
    # 解压安装包
    tar -xzvf cudnn-11.4-linux-x64-v8.2.2.26.tgz
    
    # 安装cuDNN
    sudo cp cuda/include/cudnn*.h /usr/local/cuda-11.4/include/
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.4/lib64/
    sudo chmod a+r /usr/local/cuda-11.4/include/cudnn*.h /usr/local/cuda-11.4/lib64/libcudnn*
fi

# 检查是否已经安装了Python3
if ! command -v python3 &> /dev/null
then
    # 安装Python3
    sudo apt-get install python3 python3-pip -y
fi

# 检查是否已经安装了git
if ! command -v git &> /dev/null
then
    # 安装git
    sudo apt-get install git -y
fi

# Clone stable-diffusion-webui项目
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git

# 进入stable-diffusion-webui项目目录
cd stable-diffusion-webui

# 安装所有依赖
pip3 install -r requirements.txt

# 生成一键安装stable-diffusion-webui项目的命令
echo '#!/bin/bash' > stable-diffusion-webui-install.sh
echo 'cd stable-diffusion-webui' >> stable-diffusion-webui-install.sh
echo 'python3 app.py' >> stable-diffusion-webui-install.sh
chmod +x stable-diffusion-webui-install.sh

使用步骤：

  1. 将上述脚本保存为.sh文件；
  2. 使用终端进入到保存文件所在的目录，并运行chmod +x filename.sh赋予文件可执行权限；
  3. 执行脚本./filename.sh即可安装N卡运行环境并生成一键安装stable-diffusion-webui项目的命令。