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

Posted 2023-03-27

【中文标题】如何使用 CUDA 支持构建轮文件？

【英文标题】：How to build wheel files with CUDA support?

【发布时间】：2020-09-11 04:46:42

【问题描述】：

如何创建支持 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 上运行。我怎样才能达到同样的效果？还是我的假设错了？

感谢您的帮助！

【参考方案1】：

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