cpp

    xiaoxiao2025-05-25  12

     

    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='trilinear', ext_modules=[ CUDAExtension('trilinear', [ 'src/trilinear_cuda.cpp', 'src/trilinear_kernel.cu', ]) ], cmdclass={ 'build_ext': BuildExtension }) else: print('NO CUDA is found. Fall back to CPU.') setup(name='trilinear', ext_modules=[CppExtension('trilinear', ['src/trilinear.cpp'])], cmdclass={'build_ext': BuildExtension})

    vs高版本,x64命令行下执行,

    需要先设置

    set DISTUTILS_USE_SDK=1

    最新回复(0)