Windows安装AdelaiDet的血与泪

https://blog.csdn.net/weixin_42644340/article/details/109178660icon-default.png?t=LA92https://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:\Users\Mars\Anaconda3\envs\m_space\lib\site-packages\torch\__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根目录\Lib\site-packages\torch\include\torch\csrc\jit\argument_spec.h

file1: 
  {your evn path}\Lib\site-packages\torch\include\torch\csrc\jit\argument_spec.h
  example:
  {C:\Miniconda3\envs\py36}\Lib\site-packages\torch\include\torch\csrc\jit\argument_spec.h(190)
    static constexpr size_t DEPTH_LIMIT = 128;
      change to -->
    static const size_t DEPTH_LIMIT = 128;

<2>Anaconda根目录\Lib\site-packages\torch\include\pybind11\cast.h

file2: 
  {your evn path}\Lib\site-packages\torch\include\pybind11\cast.h
  example:
  {C:\Miniconda3\envs\py36}\Lib\site-packages\torch\include\pybind11\cast.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 

  • 1
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值