树莓派4B安装PyTorch并运行YOLOv5的.pt模型问题求助
树莓派4B部署YOLOv5模型的问题解决
一、搞定.pt文件传输自动变ZIP的坑
- 直接用scp传输:本地终端(Linux/macOS)执行
scp /你本地的best.pt路径 pi@树莓派IP:/home/pi/目标存放路径;Windows可用PowerShell或者WinSCP图形化工具直接拖放.pt文件,不会触发自动压缩 - 绕开云盘自动压缩机制:要么临时把.pt后缀改成.txt,上传到云盘后再改回原后缀;要么使用云盘的「单个文件上传」功能,不要通过文件夹上传
二、解决PyTorch安装后“illegal instruction”错误
这个错误是因为你安装的PyTorch版本包含了树莓派4B不支持的CPU指令集,按以下步骤安装适配版本即可:
- 激活你的虚拟环境:
source 你的虚拟环境目录/bin/activate - 卸载现有不合规的PyTorch组件:
pip uninstall torch torchvision torchaudio -y - 安装适配aarch64架构的PyTorch版本:
优先尝试官方预编译包:
如果仍报错,换用兼容ARMv7的版本(树莓派4B向下兼容该架构):pip3 install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cpupip install torch==1.13.1+cpu torchvision==0.14.1+cpu torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cpu - 验证安装有效性:激活虚拟环境后,执行
python -c "import torch; print(torch.__version__); print(torch.cuda.is_available())",能正常输出版本号即说明安装成功 - 部署YOLOv5:克隆YOLOv5仓库到树莓派,安装依赖
pip install -r requirements.txt,最后运行检测命令python detect.py --weights /你的best.pt路径 --source 0(0代表本地摄像头,也可替换为图片/视频路径)
结论
完全可以在你的树莓派配置上部署YOLOv5模型,不存在无法实现的情况,解决上述两个核心问题即可完成部署。
内容的提问来源于stack exchange,提问作者kurtouano123
相关产品推荐
相关产品推荐

