使用TorchHub拉取YOLOv5模型时出现Runtime Error问题求助
解决M1 Mac下torch.hub加载YOLOv5报错的方案
针对你遇到的问题,给几个可行的解决办法:
手动把YOLOv5仓库放到torch缓存目录
终端里执行以下命令,直接把仓库克隆到指定缓存路径:cd /Users/myaccount/.cache/torch/hub git clone https://github.com/ultralytics/yolov5.git完成后再运行你的测试代码,就能从缓存读取仓库了。
直接用本地仓库加载模型
先把YOLOv5克隆到本地任意文件夹,然后修改代码指定本地路径加载:import torch # 把下面的路径换成你本地克隆的yolov5文件夹路径 model = torch.hub.load('/Users/yourname/Documents/yolov5', 'yolov5s', source='local', pretrained=True)加
source='local'会跳过网络拉取,直接用本地代码。验证M1 Mac的PyTorch环境
M1的ARM架构需要适配的PyTorch版本,确保你装的是支持M1的版本。如果之前装的不对,重新安装:# pip方式 pip3 install torch torchvision torchaudio # 或者conda方式(需要先装conda) conda install pytorch torchvision torchaudio -c pytorch-nightly检查系统防火墙/代理
虽然你确认网络正常,但系统防火墙或第三方安全软件可能拦截了Python的网络请求。可以暂时关闭防火墙试试,或者在系统设置里给Python开启网络权限。
内容的提问来源于stack exchange,提问作者0marA
相关产品推荐
相关产品推荐

