小记：Windows 编译 Pytorch CUDA Extension 项目

之前一直在 Windows 上编译不了，最近终于弄好了，故在此记录过程。

首先 CUDA 版本必须和安装的 Pytorch 所对应的 CUDA 版本完整对应！不能偷懒！

这里我使用的是 CUDA 12.1 的 Pytorch 2.2.1 ，所以下载 CUDA 12.1 。

原来我装的是 12.2 ，哪怕只差了一个版本也无法编译。

而且编译提供的信息也看不出是哪的问题，只能知道是在 run build_ext 这个步骤的错误。

GCC 编译器也很重要，从 Linux 上编译的教训可知，不要顶格安装该 CUDA 支持的最高编译器版本。我原先装的是 GCC 13.1 ，这个肯定不行， CUDA 12.1 最高支持 GCC 12.3 。于是参考我在 Linux 使用 GCC 11.4 ，也下了个同样版本的。

最终编译成功！！！

image-20241009110152042

报错： arch_list[-1] += '+PTX' IndexError: list index out of range

image-20241009105656921

这个是因为开了核显模式，GPU 断开连接了，导致设备列表为空。把显卡连上就能编译了。