【Network】win10+cuda11.0+pytorch1.7.1安装mmcv及swin transformer的测试

1. 创建虚拟环境,并安装pytorch

```
conda create --name TargetD python==3.7
activate TargetD
```

安装cuda11.0对应的版本:

```
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
```

2. VS 2019 Enterprise(Enterprise)

将C:\Program Files (x86)\Microsoft Visual Studio\2019\Enterprise\VC\Tools\MSVC\14.29.30133\bin\Hostx86\x64添加到系统的环境变量中。
在这里插入图片描述
测试是否成功:

C:\Users\Administrator>cl
用于 x64 的 Microsoft (R) C/C++ 优化编译器 19.29.30037 版
版权所有(C) Microsoft Corporation。保留所有权利。

用法: cl [ 选项... ] 文件名... [ /link 链接选项... ]

3. 安装 MMCV 1.3.13

在进行如下操作时,一定要进入cmd,然后activate TargetD,不要在anaconda prompt下进行。

git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
git checkout v1.3.13
pip3 install -r requirements.txt

配置环境变量:
CUDA_PATH的路径一般在安装CUDA时就自动添加了,如果没有就自己添加
在这里插入图片描述
继续在系统变量中添加如下环境变量:
变量名称:MMCV_WITH_OPS,值:1
变量名称:MAX_JOBS,值:4
变量名称:TORCH_CUDA_ARCH_LIST,值:8.0 (尤为重要)
针对8.0的解释,这个为GPU的算力值,查看算力值,方法如下:
在这里插入图片描述
在这里插入图片描述
可以看到,我的GPU算力值为8.6,由于CUDA的版本对算力值有要求,所以针对cuda11.0,将算力值设为8.0。本人测试过8.6以及7.5都失败

4. 编译安装

```
# build
python setup.py build_ext # if success, cl will be launched to compile ops
```

出现错误:

RuntimeError: Error compiling objects for extension

解决方法:
修改对应env(TargetD)下Lib\site-packages\torch\utils\cpp_extension.py

self.use_ninja = kwargs.get('use_ninja', True)
改为 
self.use_ninja = kwargs.get('use_ninja', False)

然后依次执行:

python setup.py build_ext --inplace

正确的是显示 running build_ext,然后

pip install -e .

安装成功后,继续进行环境安装:

# install
python setup.py develop

检查安装情况:

(base) H:\test\mmcv>pip show mmcv-full
Name: mmcv-full
Version: 1.3.13
Summary: OpenMMLab Computer Vision Foundation
Home-page: https://github.com/open-mmlab/mmcv
Author: MMCV Contributors
Author-email: openmmlab@gmail.com
License: UNKNOWN
Location: h:\test\mmcv
Requires: addict, numpy, packaging, Pillow, pyyaml, yapf, regex
Required-by:

5. 大功告成

在 mmcv 文件夹里会有一个 _ext.cp37-win_amd64.pyd文件生成
在这里插入图片描述

6. 测试swin-transformer

链接:https://download.csdn.net/download/qq_44747572/85115153
在这里插入图片描述

参考

swin transformer的源码:
https://github.com/SwinTransformer/Swin-Transformer-Object-Detection

mmcv的安装:

  1. https://github.com/open-mmlab/mmcv

  2. https://www.zywvvd.com/notes/environment/cuda/windows-mmcv-1-3-13-install/windows-mmcv-1-3-13-install/

  3. https://blog.csdn.net/lebusini/article/details/117525764

  4. https://blog.csdn.net/Dorisxd/article/details/121418280

  5. https://blog.csdn.net/chen_yanan/article/details/123866039

  6. https://blog.csdn.net/weixin_43678195/article/details/123574385

  • 2
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

只搬烫手的砖

你的鼓励将是我最大的动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值