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)
发布评论