如何在Raspberry pi 4b中安装pytorch并运行.pt文件？

Question

所以我正在尝试实现我的 yolov5 对象检测模型“best.pt”，我在自定义 roboflow 协作笔记本文件中进行了训练，该文件我已经下载到我的桌面中，因为它是我训练它的地方。现在我尝试使用不同的方法将 .pt 文件传递到我的 raspi 4b（谷歌驱动器、Dropbox 等），但它总是被转换为 zip 存档文件，这是不应该的。此外，在尝试了具有arm/aarch兼容性的不同版本后，我也无法在我的raspi上获得正确的pytorch版本。运行我的 python 文件会显示“非法指令”，每当我卸载显示不兼容的 pytorch 时，该指令就会再次起作用。我也使用虚拟环境来确保我的系统的安全。

我的情况真的不可能吗？这对我来说是一个关键时刻，因为这是我使用 pytorch 在论文中写的内容。

这是我的 raspi 版本和详细信息：

PRETTY_NAME="Debian GNU/Linux 12 (bookworm)"
NAME="Debian GNU/Linux"
VERSION_ID="12"
VERSION="12 (bookworm)"
VERSION_CODENAME=bookworm
ID=debian

64bit
aarch64
python 3.11.2

Answer 1

如果您收到“非法指令”，则意味着您正在尝试在树莓派上执行不兼容的二进制文件。如果你通过 pip 安装 pytorch，它应该会自动处理这个问题：

pip3 install torch torchvision torchaudio

此外，如果这些云提供商正在压缩您的模型，您仍然可以将其解压缩到您的 pi 上。如果您因某种原因确实无法压缩文件，请尝试使用scp

来移动它们。