Setting up PyTorch plugin upfirdn2d_plugin...failed!可能的解决思路

问题描述

styleGAN系列模型训练过程中出现Setting up PyTorch plugin upfirdn2d_plugin…failed!的问题。

---

原因分析：

ninja自动编译失败。

---

解决方案：

手动编译ninja。
步骤：
（1）前置条件：conda activate一个装了ninja包的环境。
（2）到/home/**/.cache/torch_extensions/**/下查看这两个文件夹：

如果bias_act_plugin或upfirdn2d_plugin中的内容不是如下的四个：（会看到只有第一个和第四个文件）

那么就是没有编译成功。

（3）在该文件夹（bias_act_plugin或upfirdn2d_plugin下），打开cmd运行命令：

ninja

即可编译。

（4）编译过程种如果遇到报错：