如何使用 CUDA 支持构建轮文件？

Question

如何创建支持 CUDA 的 .whl 车轮文件？我有一个 facebook 的 detectron2 项目的分支，我做了一些改变。现在我想建立这个项目。当我运行这个命令：python setup.py bdist_wheel 时，它会创建一个名称为：projectname-0.1.1-cp37-cp37m-linux_x86_64.whl 的 wheel 文件。但是，当我查看 facebook 的detectron2 的官方轮子网站时，轮子文件有一个独特的命名约定，即detectron2-0.1.3+cu102-cp38-cp38-linux_x86_64.whl。在这里您可以看到该项目是使用 CUDA10.2 功能构建的。我假设我的项目的轮子文件只在 CPU 上运行。我怎样才能达到同样的效果？还是我的假设错了？

感谢您的帮助！

Answer 1

detectron2includes the suffix in the version。即，版本不是0.1.3，而是0.1.3+cu102。