PyTorch 1.0 中文文档：torch.utils.cpp_extension
PyTorch 1.0 中文文档：torch.utils.cpp_extension
ApacheCN_飞龙
发布于 2022-05-07 14:21:53
发布于 2022-05-07 14:21:53
文章被收录于专栏：信数据得永生
译者: belonHan
torch.utils.cpp_extension.CppExtension(name, sources, *args, **kwargs)创建一个C++的setuptools.Extension。
便捷地创建一个setuptools.Extension具有最小（但通常是足够）的参数来构建C++扩展的方法。
所有参数都被转发给setuptools.Extension构造函数。
例子
>>> from setuptools import setup
>>> from torch.utils.cpp_extension import BuildExtension, CppExtension
>>> setup(
name='extension',
ext_modules=[
CppExtension(
name='extension',
sources=['extension.cpp'],
extra_compile_args=['-g'])),
],
cmdclass={
'build_ext': BuildExtension
})torch.utils.cpp_extension.CUDAExtension(name, sources, *args, **kwargs)为CUDA/C++创建一个setuptools.Extension。
创建一个setuptools.Extension用于构建CUDA/C ++扩展的最少参数（但通常是足够的）的便捷方法。这里包括CUDA路径，库路径和运行库。 所有参数都被转发给setuptools.Extension构造函数。
本文参与 腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。
原始发表：2019-03-07，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
腾讯云开发者
