Apple M1设备无法安装torch>=1.7.0,求适配的torch版本
M1 Mac安装YOLOv5时Torch版本适配解决方案
M1芯片属于arm64架构,PyPI默认源没有提供适配该架构的旧版Torch包,直接执行pip install -r requirements.txt会出现找不到符合版本要求的Torch的错误。以下是具体解决步骤:
先清理可能残留的Torch相关包:
pip uninstall torch torchvision torchaudio -y安装适配M1的Torch版本,推荐安装Torch 2.0.0及以上版本(既满足YOLOv5对Torch>=1.7.0的要求,又适配arm64架构):
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu注:该命令安装的CPU版本Torch已支持M1的Metal加速,完全满足YOLOv5的运行需求。
完成Torch安装后,再安装YOLOv5的其余依赖:
pip install -r requirements.txt
如果仍有问题,可以手动修改requirements.txt中的Torch版本约束,将torch>=1.7.0改为torch>=2.0.0,避免pip尝试查找不兼容的旧版本。
内容的提问来源于stack exchange,提问作者user3588569
相关产品推荐
相关产品推荐

