Stable Diffusion_Ubuntu20_py310_colab_v2.ipynb

{"cells":[{"cell_type":"code","execution_count":null,"metadata":{"id":"gst15EuV4TF2"},"outputs":[],"source":["#@title ## 1.1 操作系统环境,如断开代码,注意是否开启GPU { display-mode: \"form\" }\n","\n","#在菜单中依次选择 修改 > 笔记本设置，硬件加速器选项启用 GPU，然后重新执行该代码单元格。也可以在 代码执行程序 > 更改运行时类型，启用 GPU\"\n","\n","from google.colab import runtime\n","gpu_info = !nvidia-smi\n","gpu_info = '\\n'.join(gpu_info)\n","if gpu_info.find('failed') >= 0:\n","\n","  runtime.unassign()\n","\n","else:\n","  print(gpu_info)\n","\n","#!nvidia-smi # 文件/笔记本设置/GPU开启\n","!sudo apt-get update\n","!sudo apt install build-essential zlib1g-dev libncurses5-dev libgdbm-dev libnss3-dev libssl-dev libreadline-dev libffi-dev wget\n","!sudo apt install aria2"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"3sIs70t8PmuK"},"outputs":[],"source":["#@title ## 1.2 下载已编译版 python 3.10.9\n","%cd /content\n","!wget https://github.com/lymanzhao/StableDiffusion_py310_colab/releases/download/1.0/Python-3.10.9.tar.gz\n","#!rm -rf Python-3.10.10\n","#!mkdir Python-3.10.10\n","!tar -x Python-3.10.9 -zxvf Python-3.10.9.tar.gz\n"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"ZMsg-_FgOEqr"},"outputs":[],"source":["#@title ## [可选]1.2 Python310源码编译 { display-mode: \"form\" }\n","# !wget https://www.python.org/ftp/python/3.10.9/Python-3.10.9.tgz\n","# !tar -xf Python-3.10.9.tgz\n","# %cd Python-3.10.9/\n","# !./configure --enable-optimizations\n","# !make -j 8\n","# # 压缩源码，上传到github\n","# %cd /content\n","# !tar -zcvf Python-3.10.9.tar.gz Python-3.10.9"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"aRIiIDw45SOs"},"outputs":[],"source":["#@title ## 1.3 Python310环境配置 { display-mode: \"form\" }\n","\n","%cd /content/Python-3.10.9\n","!sudo make altinstall\n","\n","!which python3\n","!which python3.10\n","!sudo update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.8 1\n","!sudo update-alternatives --install /usr/bin/python3 python3 /usr/local/bin/python3.10 2\n","#!sudo update-alternatives --config python3\n","!sudo update-alternatives  --set python3 /usr/local/bin/python3.10\n","# 选择2使用Python 310版本\n","!python -V\n","!curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py\n","!python3 get-pip.py"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"JZnASWq_Ou4T"},"outputs":[],"source":["#@title ## 2.1 下载安装webui { display-mode: \"form\" }\n","%cd /content\n","!git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git\n","%cd /content/stable-diffusion-webui\n","!COMMANDLINE_ARGS=\"--exit\" REQS_FILE=\"requirements.txt\" python launch.py\n"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"6M0AhP8gzEV1"},"outputs":[],"source":["#@title ## 2.2 下载常用插件{ display-mode: \"form\" }\n","%cd /content/stable-diffusion-webui\n","!wget -N https://raw.githubusercontent.com/lymanzhao/StableDiffusion_py310_colab/main/SDextensionsDownload.py\n","!python SDextensionsDownload.py"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"FkTA-omHGlk_"},"outputs":[],"source":["#@title ## 2.3 下载模型{ display-mode: \"form\" }\n","%cd /content/stable-diffusion-webui\n","\n","!mkdir -p /content/stable-diffusion-webui/models/Lora\n","!wget -N https://raw.githubusercontent.com/lymanzhao/StableDiffusion_py310_colab/main/SDModelDownload.py\n","!python SDModelDownload.py\n"]},{"cell_type":"code","source":["#@title # 2.4 [可选]下载安装controlnet相关，不用的情况下可跳过\n","%cd /content/stable-diffusion-webui\n","\n","!wget -N https://raw.githubusercontent.com/lymanzhao/StableDiffusion_py310_colab/main/SDcontrolnetExtensionsDownload.py\n","!python SDcontrolnetExtensionsDownload.py"],"metadata":{"id":"jWcia-Nb35j8"},"execution_count":null,"outputs":[]},{"cell_type":"code","execution_count":null,"metadata":{"id":"ivRg827YUdyK"},"outputs":[],"source":["#@title # 3 运行\n","%cd /content/stable-diffusion-webui\n","!hostname -I\n","#!COMMANDLINE_ARGS=\"--xformers --deepdanbooru --share --gradio-debug --gradio-auth zyl:2023\" REQS_FILE=\"requirements.txt\" python launch.py #注意加密登录参数\n","!COMMANDLINE_ARGS=\"--xformers --no-half --share --gradio-debug\" REQS_FILE=\"requirements.txt\" python launch.py"]},{"cell_type":"code","source":["#@title # 4 训练\n","\n","## 安装脚本\n","%cd /content/stable-diffusion-webui\n","\n","!wget -N https://raw.githubusercontent.com/lymanzhao/StableDiffusion_py310_colab/main/SDTrainingExtensionsDownload.py\n","!python SDTrainingExtensionsDownload.py\n","\n","## 导入图片\n","#from google.colab import drive\n","# wget https://drive.google.com/u/0/uc?id=1-0J2bK19otMR4NQD2hwKjN58irTPeWWk&export=download\n","#drive.mount('/content/gdrive/', force_remount=True)\n","#!unzip -j -o /content/gdrive/MyDrive/gwei.zip -d /content/gwei/trainimage\n"],"metadata":{"id":"yEL0aXFZ7e3_"},"execution_count":null,"outputs":[]},{"cell_type":"code","source":[],"metadata":{"id":"9PWBciWF53TY"},"execution_count":null,"outputs":[]}],"metadata":{"accelerator":"GPU","colab":{"provenance":[{"file_id":"1Mgd2yhWsSH1-5PH8b3OyD6UxYhBqZF9m","timestamp":1678686074331}],"private_outputs":true,"authorship_tag":"ABX9TyMr0RzfXSF07x2yyywuSCaY"},"gpuClass":"standard","kernelspec":{"display_name":"Python 3","name":"python3"},"language_info":{"name":"python"}},"nbformat":4,"nbformat_minor":0}