windows安装mmcv

最新推荐文章于 2024-07-25 19:46:27 发布

张占坤

最新推荐文章于 2024-07-25 19:46:27 发布

阅读量886

点赞数 1

文章标签： windows

本文链接：https://blog.csdn.net/m0_46819545/article/details/132866015

版权

1、首先根据这个教程安装GPU版本的Pytorch及torchvision

安装GPU版本的Pytorch及torchvision

2、从 Windows 菜单启动 Anaconda 命令行

不要使用原始的 cmd.exe 或是 powershell.exe。命令行有两个版本，一个基于 PowerShell，一个基于传统的 cmd.exe。请注意以下说明都是使用的基于 PowerShell
在这里插入图片描述

3、进入虚拟环境（以下操作均在虚拟环境里进行）

conda activate 环境名

4、设置 MSVC 编译器

设置环境变量。添加 C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.27.29110\bin\Hostx86\x64 到 PATH
然后在命令行中输入set Path=C可以让环境变量的修改立马生效
保存之后重新打开命令行
输入cl，若是显示出了19的编译器，那么就证明没有问题了
在这里插入图片描述

5 直接安装简单快捷

在这里选择要安装的版本
在这里插入图片描述
然后复制到虚拟环境终端就行了
如果这里成功了下面就不用看了，这是两个办法

6 源码安装

6.1、克隆代码仓库

git clone https://github.com/open-mmlab/mmcv.git
cd mmcv

6.2、安装 ninja 和 psutil 以加快编译速度

pip install -r requirements/optional.txt

6.3、设置 CUDA 的目标架构

# 这里需要改成你的显卡对应的目标架构
$env:TORCH_CUDA_ARCH_LIST="7.5"
$env:MMCV_WITH_OPS = 1
$env:MAX_JOBS = 8（计算机不好的话改成4）

可以通过这个exe文件查看显卡算力C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite\deviceQuery.exe"
Device 0: "NVIDIA GeForce GTX 1660 SUPER"
CUDA Driver Version / Runtime Version          11.7 / 11.1
CUDA Capability Major/Minor version number:    7.5（这就是上面为什么是7.5）

在这里插入图片描述

6.4、编译安装

python setup.py build_ext  # 如果成功, cl 将被启动用于编译算子