cpp_extension dist must be a Distribution instance

最新推荐文章于 2023-11-15 14:47:52 发布

原创最新推荐文章于 2023-11-15 14:47:52 发布 · 1.6w 阅读

收录于

当前文章被以下社区和专栏收录：

python宝典专栏收录该内容

1239 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

这篇博客探讨了在C++扩展中遇到的'dist must be a Distribution instance'错误。问题源于传递给函数的参数不正确，应当为Distribution实例。文章提到了在VS高版本和x64命令行环境下，通过设置DISTUTILS_USE_SDK=1环境变量尝试解决该问题，但指出两种常见写法都存在错误。

dist must be a Distribution instance

原因是参数传的不对参数应该是Distribution 实例

有两种写法，但是都不对

ffi = cpp_extension.build_ext(

ffi = cpp_extension.BuildExtension(
    '_ext.dcn_v2',
    headers=headers,
    sources=sources,
    define_macros=defines,
    relative_to=__file__,
    with_cuda=with_cuda,
    extra_objects=extra_objects,
    # extra_compile_args=extra_compile_args
)

c++用法：

from setuptools import setup
import torch
from torch.utils.cpp_extension import BuildExtension, CUDAExtension, CppExtension

if torch.cuda.is_available():
    print('Including CUDA code.')
    setup(
        name='trilinea

了解本专栏