Win10安装Detectron2(0.1或0.4.1,基于Pytorch1.8.1、Cuda11.1.1)

编程入门 行业动态 更新时间:2024-10-20 20:32:31

Win10安装Detectron2(0.1或0.4.1,基于Pytorch1.8.1、Cuda11.1.1)

Win10安装Detectron2(0.1或0.4.1,基于Pytorch1.8.1、Cuda11.1.1)

Win10安装Detectron2(0.1或0.4.1,基于Pytorch1.8.1、Cuda11.1.1)

1.安装cuda和对应的cudnn

2.安装vs2019

3.安装pytorch

4.安装pycoco

5.安装fvcore

6.安装Detectron2

6.1.版本0.1(老版本)

6.2.版本0.4.1(新版本)

FAQ     

1.出现UserWarning: It seems that the VC environment is activated but DISTUTILS_USE_SDK is not set.This may lead to multiple activations of the VC env.Please set `DISTUTILS_USE_SDK=1` and try again.

2.出现ModuleNotFoundError: No module named ‘win32con‘和No module named ‘antlr4‘


  ⚡插播一条老家自产的糖心苹果,多个品种,欢迎选购!有问题随时私信我⚡:

🍎🍎来自雪域高原的馈赠——海拔2000米的大凉山高原生态糖心苹果🍎🍎

1.安装cuda和对应的cudnn

        这部分略过,可以参考我的这篇文章:

WIN10安装多个版本的cuda及cudnn并自动切换_Jayce~的博客-CSDN博客=1001.2014.3001.5501

        安装的时候需要查看detectron2的环境要求:Installation — detectron2 0.6 documentation.html        我这边直接安装最新的cuda11.1.1+cudnn8.2.1+torch1.8:

        另外,为了保证环境的简洁,最好新建一个虚拟环境安装~~ 

2.安装vs2019

        因为是自己编译,需要借助vs2019的编译器,直接下载社区版即可:感谢您下载 Visual Studio - Visual Studio/?sku=Community&rel=16        然后在安装的时候,选择安装一个工作负荷以提供编译器:

        安装完成后,使用everything搜索系统中是否存在vcvars64.bat文件,如果有的话直接CMD执行call命令初始化,然后输入cl验证是否安装成功:

3.安装pytorch

        直接去pytorch官网:PyTorch/        选择合适版本的pytorch,并在CMD中执行官网提供的pip命令即可:

安装完成后验证是否安装成功:

4.安装pycoco

        直接使用git从 下载到本地,并切换到cocoapi\PythonAPI文件夹下面,然后执行如下命令安装:

python setup.py build_ext --inplace
python setup.py build_ext install

如果第一句话报错没有cython,就需要安装一下cython:

安装完成后,导入如果没有出错则安装成功:

5.安装fvcore

        直接使用git从下述网站clone到本地:://github/facebookresearch/fvcore        并切换到保存路径夹下面找到setup.py文件,然后执行如下命令安装:

python setup.py build --force develop

6.安装Detectron2

6.1.版本0.1(老版本)

        直接使用git从下述网站clone到本地:://github/conansherry/detectron2        然后修改如下的几个文件:

1.cpp_extension.py中的decode()改为decode(' .gbk')

2.argument_spec.h文件中的static const size_t ARG_SPEC_DEPTH_LIMIT = 128改为static constexpr size_t ARG_SPEC_DEPTH_LIMIT = 128

3.将deform_conv.h和deform_conv_cuda.cu文件中的AT_CHECK全局替换为TORCH_CHECK

4.切换到detectron2的保存路径,执行python setup.py build --force develop安装,并验证

6.2.版本0.4.1(新版本)

        直接使用git从下述网站clone到本地:://github/facebookresearch/detectron2        然后修改如下的几个文件:

1.cpp_extension.py中的decode()改为decode(' gbk')

2.ROIAlignRotated_cuda.cu文件将ceil替换为ceilf(注意是全词匹配)

3.deform_conv_cuda_kernel.cu文件将floor替换为floorf(注意是全词匹配)

4.nms_rotated_cuda.cu文件将下图红框中删除

并添加下方红框中的代码:

5.切换到detectron2的保存路径,执行python setup.py build --force develop安装,并验证

FAQ     

1.出现UserWarning: It seems that the VC environment is activated but DISTUTILS_USE_SDK is not set.This may lead to multiple activations of the VC env.Please set `DISTUTILS_USE_SDK=1` and try again.

        只需要执行如下命令,指定一下即可:

SET DISTUTILS_USE_SDK=1
call "C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Auxiliary\Build\vcvars64.bat"

        然后重新执行python setup.py build --force develop。

2.出现ModuleNotFoundError: No module named ‘win32con‘和No module named ‘antlr4‘

Win10出现ModuleNotFoundError: No module named ‘win32con‘和No module named ‘antlr4‘

都看到这里了,还不赶紧点赞评论收藏走起?要是能打赏作者一波就更香了!

更多推荐

Win10安装Detectron2(0.1或0.4.1,基于Pytorch1.8.1、Cuda11.1.1)

本文发布于:2024-02-26 09:01:04,感谢您对本站的认可!
本文链接:https://www.elefans.com/category/jswz/34/1701954.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:

发布评论

评论列表 (有 0 条评论)
草根站长

>www.elefans.com

编程频道|电子爱好者 - 技术资讯及电子产品介绍!