disco_local

Disco_Diffusion

Disco Diffusion 是发布于 Google Colab 平台的一款利用人工智能深度学习进行数字艺术创作的工具，它是基于 MIT 许可协议的开源工具，可以在 Google Drive 直接运行。我是将其部署在腾讯云服务器进行运行的，然后速度相对较快一点。

部署流程

1. 购买一台gpu服务器 必须是英伟达显卡，AMD不支持

• 建议配置 gpu 16g+
• 我本地的使用情况

2. 安装 NVIDIA 驱动、cuda、cudnn

• 腾讯云参考
• 本系统采用腾讯云镜像市场安装方式 版本如下

3. 进入服务器 (但腾讯云服务器下载外网资源很慢 所以这种方式一般会执行失败) 国内可采用第四步代替

   git clone https://github.com/alembics/disco-diffusion.git
   // 在 docker/prep 目录下
   docker build -t disco-diffusion-prep:5.1 .
   // 在 docker/main 目录下
   docker build -t disco-diffusion:5.1 .
   docker run --rm -it \
       -v $(echo ~)/disco-diffusion/images_out:/workspace/code/images_out \
       -v $(echo ~)/disco-diffusion/init_images:/workspace/code/init_images \
       --gpus=all \
       --name="disco-diffusion" --ipc=host \
       --user $(id -u):$(id -g) \
   disco-diffusion:5.1 python disco-diffusion/disco.py

4. 克隆我仓库下的版本 对外网东西做了本地化修改

   git clone https://gitee.com/lizhongyuan3/disco-diffusion.git
   // 在 docker/prep 目录下
   docker build -t disco-diffusion-prep:5.1 .
   // 在 docker/main 目录下
   docker build -t disco-diffusion:5.1 .
   docker run --rm -it \
       -v $(echo ~)/disco-diffusion/images_out:/workspace/code/images_out \
       -v $(echo ~)/disco-diffusion/init_images:/workspace/code/init_images \
       -v /root/disco-diffusion/disco.py:/workspace/code/disco-diffusion/disco.py \
       --gpus=all \
       --name="disco-diffusion" --ipc=host \
       --user disco:disco \
   disco-diffusion:5.1 python disco-diffusion/disco.py

   资源包

   资源包下载请联系我 分享给你阿里网盘连接