1,树莓派安装pytorch(64位)
进行下面步骤之前一定要先更新numpy,更新方法在上一篇文章中有写到
1.1安装torch
不可以安装最新版本,最新版本有bug,推荐安装1.8.1版本
pip install torch==1.8.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
1.2安装torchvision
,注意版本要对应
pip install torchvision==0.9.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
2,树莓派安装pytorch(32位)
相对于64位安装比较麻烦,需要手动下载wheel文件然后安装
推荐文章:32位树莓派 安装pytorch
以我的树莓派为例
把这两个wheel文件下载到树莓派桌面上
然后cd Desktop
2.1安装torch
pip3 install torch-1.8.1-cp39-cp39m-linux_armv7l.whl(自己刚下载的文件名)
验证:
python3
>>> import torch
>>> torch.rand(5,3)
2.2 安装torchvison
安装依赖
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev
sudo apt-get install libavcodec-dev libavformat-dev libswscale-dev
安装wheel文件
pip3 install 自己下载的文件名
验证
python3
>>> import torchvision
>>> torchvision.get_image_backend()
>>> torchvision.get_video_backend()
3.安装yolov8
当对于pc不同的是不能直接运行 pip install ultralytics,要在github下载源码到树莓派上,然后将你的代码和训练好的模型放到下载的文件夹下,然后安装yolo所需的依赖,就可以运行啦,不论是v8还是v5,都可以这样操作。
yolov8官方链接:v8
yolov5官方链接:v5
这里我以v8做例子