直接pip install 会报错

Failed to build mmcv
ERROR: Could not build wheels for mmcv, which is required to install pyproject.toml-based projects
因为mmcv需要和pytorch cuda版本匹配

进入官网Installation — mmcv 2.2.0 文档

在终端输入查看自己的版本:

python -c 'import torch;print(torch.__version__);print(torch.version.cuda)'

官网上选择自己的版本,有对应的指令可以安装

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐