概述
https://blog.csdn.net/weixin_42644340/article/details/109178660https://blog.csdn.net/weixin_42644340/article/details/109178660参考了上面的博客,安装detection2,安装了两台电脑,一台dell,一台联想,两个笔记本都带显卡,现在以联想安装记录过程。所涉及的安装包我都会一并上传到网盘,分享。
0.首先确定NVIDIA显卡驱动是否安装,若安装成功输入
nvidia-smi
会显示显卡信息,如图
安装成功以后,需要安装nvidia深度学习两件套,cuda和cudnn,注意,版本有限制,良心警告,cuda版本为10.1,cudnn版本与cuda匹配即可,安装成功,输入
nvcc -V
查看cuda版本信息,安装默认即可。
上述安装完,就可以安装torch了,官方制定的版本为torch1.3,所以这里仍然得注意版本问题,之前安装dell的时候吃过亏。
1.pytorch的安装,建议直接在国内镜像中搜索下载,要不又是小半天的时间。
还有一点是,Python的大版本限定为python3.6,小版本要高于3.6.1,我的安装得Anaconda的python版本是3.6.2的,所以创建虚拟环境
conda create -n m_space python=3.6.5
比如我选择的是pytorch-1.3.0-py3.6_cuda101_cudnn7_0.tar.bz2,利用conda的离线安装命令
conda install --offline pytorch-1.3.0-py3.6_cuda101_cudnn7_0.tar.bz2
安装成功后,测试发现问题
>>> 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解决了问题
conda install numpy pyyaml mkl cmake cffi
重新测试后,成功!安装torchvision,版本最好为0.4.1
2.安装vs2019
注意勾选MSVC 功能,如图
安装完成后,准备工作结束,开始正式安装。
3.安装cocoapi
进入python相关文件夹,先输入
python setup.py build_ext --inplace
一般不会报错,报错的话就是编译器的问题,比如我在dell安装的时候让我安装vs2015去编译
再输入
python setup.py build_ext install
还有一个问题是/Wno-cpp无效的参数
可以将setup.py中相应字段删除,删完如下
from 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的目录,输入
python setup.py build --force develop
同理安装
5.安装detection2
根据官方的提示,需要修改相应的文件,如果是虚拟环境,则定位到虚拟环境的安装位置,而非Anaconda根目录
<1> Anaconda根目录Libsite-packagestorchincludetorchcsrcjitargument_spec.h
file1:
{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
file2:
{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); }
修改完之后,输入
python setup.py build develop
即可安装
6.安装AdelaiDet
同样进入到安装目录,输入
python setup.py build develop
有缺失包,可以终止安装,利用pip进行安装
7.测试
下载好权重,比如R_101_3x.pth准备一张照片,放到根目录下,命名为01.jpg
进入AdelaiDet的根目录,输入
python 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进行安装
pip 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的血与泪所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
发表评论 取消回复