【Network】win10+cuda11.0+pytorch1.7.1安装mmcv

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

```
conda create --name TargetD==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添加到系统的环境变量中。
【Network】win10+cuda11.0+pytorch1.7.1安装mmcv
测试成功:

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时就自动添加了,如果没有就自己添加
【Network】win10+cuda11.0+pytorch1.7.1安装mmcv
继续在系统变量中添加以下环境变量:
变量名称:MMCV_WITH_OPS,值:1
变量名称:MAX_JOBS,值:4
变量名称:TORCH_CUDA_ARCH_LIST,值:8.0 (尤为重要)
针对8.0的解释,这个为GPU的算力值,查看算力值,方法如下:
【Network】win10+cuda11.0+pytorch1.7.1安装mmcv
【Network】win10+cuda11.0+pytorch1.7.1安装mmcv
可以看到,我的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文件生成
【Network】win10+cuda11.0+pytorch1.7.1安装mmcv

6. 测试swin-transformer

链接:https://download.csdn.net/download/qq_44747572/85115153
【Network】win10+cuda11.0+pytorch1.7.1安装mmcv

参考

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

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
xiaoxingxing的头像xiaoxingxing管理团队
上一篇 2022年4月13日 下午6:38
下一篇 2022年4月13日 下午6:55

相关推荐