https://blog.csdn.net/weixin_42644340/article/details/109178660https://blog.csdn.net/weixin_42644340/article/details/109178660参考了上面的博客,安装detection2,安装了两台电脑,一台dell,一台联想,两个笔记本都带显卡,现在以联想安装记录过程。所涉及的安装包我都会一并上传到网盘,分享。
0.首先确定NVIDIA显卡驱动是否安装,若安装成功输入
1nvidia-smi
会显示显卡信息,如图
安装成功以后,需要安装nvidia深度学习两件套,cuda和cudnn,注意,版本有限制,良心警告,cuda版本为10.1,cudnn版本与cuda匹配即可,安装成功,输入
1
2nvcc -V
查看cuda版本信息,安装默认即可。
上述安装完,就可以安装torch了,官方制定的版本为torch1.3,所以这里仍然得注意版本问题,之前安装dell的时候吃过亏。
1.pytorch的安装,建议直接在国内镜像中搜索下载,要不又是小半天的时间。
还有一点是,Python的大版本限定为python3.6,小版本要高于3.6.1,我的安装得Anaconda的python版本是3.6.2的,所以创建虚拟环境
1conda create -n m_space python=3.6.5
比如我选择的是pytorch-1.3.0-py3.6_cuda101_cudnn7_0.tar.bz2,利用conda的离线安装命令
1conda install --offline pytorch-1.3.0-py3.6_cuda101_cudnn7_0.tar.bz2
安装成功后,测试发现问题
1
2
3
4
5
6>>> import torch Traceback (most recent call last): File "<stdin>", line 1, in <module> File "C:UsersMarsAnaconda3envsm_spacelibsite-packagestorch__init__.py", line 81, in <module> from torch._C import * ImportError: DLL load failed: 找不到指定的模块
升级pip,利用pip升级numpy后仍然报错,但是利用conda解决了问题
1conda install numpy pyyaml mkl cmake cffi
重新测试后,成功!安装torchvision,版本最好为0.4.1
2.安装vs2019
注意勾选MSVC 功能,如图
安装完成后,准备工作结束,开始正式安装。
3.安装cocoapi
进入python相关文件夹,先输入
1python setup.py build_ext --inplace
一般不会报错,报错的话就是编译器的问题,比如我在dell安装的时候让我安装vs2015去编译
再输入
1python setup.py build_ext install
还有一个问题是/Wno-cpp无效的参数
可以将setup.py中相应字段删除,删完如下
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28from setuptools import setup, Extension import numpy as np # To compile and install locally run "python setup.py build_ext --inplace" # To install library to Python site-packages run "python setup.py build_ext install" ext_modules = [ Extension( 'pycocotools._mask', sources=['../common/maskApi.c', 'pycocotools/_mask.pyx'], include_dirs = [np.get_include(), '../common'], extra_compile_args=[ '-std=c99'], ) ] setup( name='pycocotools', packages=['pycocotools'], package_dir = {'pycocotools': 'pycocotools'}, install_requires=[ 'setuptools>=18.0', 'cython>=0.27.3', 'matplotlib>=2.1.0' ], version='2.0', ext_modules= ext_modules )
剩下基本上就没什么问题了,如果报找不到别的包,可以终止安装,利用pip先把包安装完
4.安装fvcore
进入setup.py的目录,输入
1python setup.py build --force develop
同理安装
5.安装detection2
根据官方的提示,需要修改相应的文件,如果是虚拟环境,则定位到虚拟环境的安装位置,而非Anaconda根目录
<1> Anaconda根目录Libsite-packagestorchincludetorchcsrcjitargument_spec.h
1
2
3
4
5
6
7file1: {your evn path}Libsite-packagestorchincludetorchcsrcjitargument_spec.h example: {C:Miniconda3envspy36}Libsite-packagestorchincludetorchcsrcjitargument_spec.h(190) static constexpr size_t DEPTH_LIMIT = 128; change to --> static const size_t DEPTH_LIMIT = 128;
<2>Anaconda根目录Libsite-packagestorchincludepybind11cast.h
1
2
3
4
5
6
7file2: {your evn path}Libsite-packagestorchincludepybind11cast.h example: {C:Miniconda3envspy36}Libsite-packagestorchincludepybind11cast.h(1449) explicit operator type&() { return *(this->value); } change to --> explicit operator type&() { return *((type*)this->value); }
修改完之后,输入
1python setup.py build develop
即可安装
6.安装AdelaiDet
同样进入到安装目录,输入
1python setup.py build develop
有缺失包,可以终止安装,利用pip进行安装
7.测试
下载好权重,比如R_101_3x.pth准备一张照片,放到根目录下,命名为01.jpg
进入AdelaiDet的根目录,输入
1python demo/demo.py --config-file configs/BlendMask/R_101_3x.yaml --input 01.jpg --opts MODEL.WEIGHTS R_101_3x.pth
报错,没有_get_trace_graph这个函数,这个是小问题,只需到文件中,将第一个下划线去掉
函数名变成get_trace_graph即可。再次输入命令,如果提示没有cv2这个包,同样可以利用pip进行安装
1pip install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-python
安装完,即可显示
完结
8.总结
cuda 10.1
python 3.6.5
torch 1.3
torchvision 0.4.0
安装包链接
baidu云盘
提取码:ugvp
最后
以上就是甜美学姐最近收集整理的关于Windows安装AdelaiDet的血与泪的全部内容,更多相关Windows安装AdelaiDet内容请搜索靠谱客的其他文章。
发表评论 取消回复