win10下detectron2的安装教程1. 安装所需基础环境2. win10下安装过程

90 阅读 0 评论 60 点赞

我是靠谱客的博主包容火，这篇文章主要介绍win10下detectron2的安装教程1. 安装所需基础环境2. win10下安装过程，现在分享给大家，希望可以做个参考。

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行，修改为 (其中注释掉的为原来的代码)：

复制代码

1
2
3
4
5
6
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前面有空格)

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
  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安装的准备工作完成，下一步直接安装即可