Building wheels for collected packages: mmcv, mmcv-full 卡住

news2024/11/15 23:53:14

安装 anime-face-detector 的时候遇到一个问题：Installation takes forever #1386：在构建mmcv-full时卡住，这里分享下解决方法（安装 mmcv 同理，将下面命令中的 mmcv-full 替换成 mmcv）

具体表现如下：

如果需要指定版本，修改命令中的 mmcv-full：mmcv-full=={mmcv_version}
例：anime-face-detector 需要的版本 >= 1.3.8, <=1.4.0

... install mmcv-full==1.3.8

方法1

官方文档推荐方法

pip install -U openmim
mim install mmcv-full

方法2

命令行

根据官方文档，获取对应的CUDA和PyTorch版本进行下载（Python版本在下面）：

# 获取CUDA版本
cuda_version=$(python3 -c "import torch; print(torch.version.cuda.replace('.', ''))")

# 获取PyTorch版本
torch_version=$(python3 -c "import torch; print(torch.__version__.split('+')[0].replace('.', ''))")

# 构造安装命令
install_command="pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu${cuda_version}/torch${torch_version}/index.html"

# 检查安装命令
echo "安装命令: ${install_command}"

# 执行安装命令
eval ${install_command}

Python版本

import torch
import subprocess

# 获取当前的CUDA和PyTorch版本
cuda_version = torch.version.cuda.replace('.', '')
torch_version = torch.__version__.split('+')[0].replace('.', '')

# 构造安装命令
install_command = f"pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu{cuda_version}/torch{torch_version}/index.html"

# 打印并执行安装命令
print("安装命令:", install_command)
subprocess.check_call(install_command, shell=True)