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



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

感谢您的帮助!

detectron2

在版本中包含后缀。 即,版本不是0.1.3而是0.1.3+cu102

最新更新