github.com/XingangPan/…

这个项目预热了这么久，没有人不知道吧，现在终于开源了。

Star20.3k，热度超高，今天准备安装试试看。

Clone仓库

 git clone https://github.com/XingangPan/DragGAN.git
 

Mac的特殊处理

按照官方的说法，默认是支持CUDA，如果使用MAC的M1\M2芯片，需要运行下面脚本。

cat environment.yml | \
  grep -v -E 'nvidia|cuda' > environment-no-nvidia.yml && \
    conda env create -f environment-no-nvidia.yml
conda activate stylegan3

export PYTORCH_ENABLE_MPS_FALLBACK=1

在这里遇到了第一个坑，我安装的conda，没有scipy1.11.0库，而在environment.yml中设置的scipy使用1.11.0库，我在 conda update --all 多次后，看到最高仍然是1.10.1，所以需要修改scipy库版本。

name: stylegan3
channels:
  - pytorch
  - nvidia
dependencies:
  - python >= 3.8
  - pip
  - numpy>=1.25
  - click>=8.0
  - pillow=9.4.0
  - scipy=1.10.1
  # - scipy=1.11.0
  - pytorch>=2.0.1
  - torchvision>=0.15.2
  - cudatoolkit=11.1
  - requests=2.26.0
  - tqdm=4.62.2
  - ninja=1.10.2
  - matplotlib=3.4.2
  - imageio=2.9.0
  - pip:
    - imgui==2.0.0
    - glfw==2.6.1
    - gradio==3.35.2
    - pyopengl==3.1.5
    - imageio-ffmpeg==0.4.3
    # pyspng is currently broken on MacOS (see https://github.com/nurpax/pyspng/pull/6 for instance)
    - pyspng-seunglab

更改后，运行上面命令成功。创建了stylegan3的环境。

下载数据集

sh scripts/download_model.sh

没什么要说的，注意点是需要魔法，这可真神奇。

运行 DragGAN GUI

要启动 DragGAN GUI，只需运行：

sh scripts/gui.sh

又出现了一个问题：

这就很不理解了，明明是按照步骤一步一步来的，为什么少了click服务。

运行：

conda install click

重新启动。

离谱，缺少pytorch包，接下来估计还会有各种这样问题，怀疑是从scipy开始就有问题了，但是检查create命令时的日志，没有看到报错信息，出现类似问题找GPT一一解决，不再赘述。

终于编译成功，到了运行阶段，又又又报错了。

重新打开还是报错。

成了

换了个高配点电脑，重复上面步骤，成了。