1. 安装所需基础环境

我所用的前置环境如下

• cuda11.1
• pytorch1.7
• python3.7
• VS2019， 参考https://blog.csdn.net/weixin_42990464/article/details/109994779
• gcc 6.3.0，参考https://blog.csdn.net/weixin_42990464/article/details/110109988
• ninja 1.10.0
• pycocotools，参考https://www.jianshu.com/p/8658cda3d553

2. win10下安装过程

2.1 首先需要先将detectron2 clone到本地

git clone https://github.com/facebookresearch/detectron2.git

2.2 修改pytorch代码

文件位置位于{your env path}Libsite-packagestorchincludetorchcsrcjitruntimeargumenta_spec.h，代码改动位置应为161行，修改为 (其中注释掉的为原来的代码)：

namespace {
//static constexpr size_t ARG_SPEC_DEPTH_LIMIT = 128;
    static const size_t ARG_SPEC_DEPTH_LIMIT = 128;
}

注意

如果你使用的是中文版本的win10，还需要额外添加一步修改：

打开你环境下的cpp_extension.py, 找到如下代码段的位置，在decode() 的括号中添加 ’ .gbk’ (注意.gbk前面有空格)

  if sys.platform.startswith('darwin'):
        # There is no particular minimum version we need for clang, so we're good here.
        return True
    try:
        if sys.platform.startswith('linux'):
            minimum_required_version = MINIMUM_GCC_VERSION
            version = subprocess.check_output([compiler, '-dumpfullversion', '-dumpversion'])
            version = version.decode(' gbk').strip().split('.')
        else:
            minimum_required_version = MINIMUM_MSVC_VERSION
            compiler_info = subprocess.check_output(compiler, stderr=subprocess.STDOUT)
            match = re.search(r'(d+).(d+).(d+)', compiler_info.decode(' gbk').strip())
            version = (0, 0, 0) if match is None else match.groups()
    except Exception:
        _, error, _ = sys.exc_info()
        warnings.warn('Error checking compiler version for {}: {}'.format(compiler, error))
        return False

2.3 进行安装

• conda activate {your env}
• call “C:Program Files (x86)Microsoft Visual Studio2019CommunityVCAuxiliaryBuildvcvars64.bat”
  此处解决的问题为，如果不实现call一下本地的vs，会造成之后的安装出现编译器错误
• 至此，win10下detectron安装的准备工作完成，下一步直接安装即可

cd detectron2
python setup.py build develop

• 安装完成后可使用如下代码，验证是否安装成功

最后

以上就是包容火为你收集整理的win10下detectron2的安装教程1. 安装所需基础环境2. win10下安装过程的全部内容，希望文章能够帮你解决win10下detectron2的安装教程1. 安装所需基础环境2. win10下安装过程所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。