在树莓派5上使用pytroch进行模型训练—全流程笔记

在树莓派上运行pytroch模型🚀

在完成了树莓派的一系列基础配置学习之后,按照规划,下一步要做的就是在树莓派上安装一个pytorch,尝试运行一下深度学习的模型,如果可以实现且准速度有一定保证的话,就可以作为一个项目方案,或者来说参加比赛用的一个非常好的选择方案,也是打算抽时间写一篇类似的东西,今天终于是完成了。



系列文章1:树莓派初探---给树莓派5 安装Raspberry Pi OS 操作系统
系列文章2:使用VNC-viewer对树莓派5 远程连接桌面---详细记录笔记版
系列文章3:树莓派5环境配置笔记 新建虚拟python环境---安装第三方库---配置Thonny解释器
系列文章4:在树莓派5上调用通义千问API实现多轮对话和流式输出


文章目录

1.在树莓派上新建一个python的虚拟环境并安装pytroch(CPU)


在进行pytroch任务之前可以先建一个虚拟环境。

如果是纯新手的的话可以去看一下更详细的这个入门版教程:树莓派5环境配置笔记 新建虚拟python环境---安装第三方库---配置Thonny解释器

新建虚拟环境

cmd 复制代码
python3 -m venv pytroch

激活虚拟环境

cmd 复制代码
source pytroch/bin/activate

配置清华园镜像

cmd 复制代码
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

更新软件下载仓库的索引文件

cmd 复制代码
sudo apt update

安装pytroch,树莓派上没有GPU所以我们直接pip安装就行安装一个CPU的版本。

cmd 复制代码
pip3 install torch torchvision torchaudio

安装成功之后看下pytorch的版本。

启动python

cmd 复制代码
python

导入torch

cmd 复制代码
import torch

查看pytroch版本

cmd 复制代码
troch.__version__

退出python解释器

cmd 复制代码
exit()

1+.安装matplotlib绘图库(之后绘制损失曲线用的到)


进入到之前建好的pytroch环境下,然后直接安装就行。

cmd 复制代码
pip3 install matplotlib

进入到环境中之后直接安装就行

2.使用的训练代码(仓库/详解博客)


我使用的代码是我自己之前写的基于一维卷积神经网络的深度学习训练模板代码,自己下载一下就行了,用自己的也行。

GitHub仓库地址:https://github.com/StChenHaoGitHub/1D_Pytorch_Train_demo

博客地址:https://chen-hao.blog.csdn.net/article/details/135992162

3.通过U盘传输模型代码


将代码下在U盘里然后直接插入,这里为了避免远程传输可能出现的各种风险,先直接用硬件传输代码。

插入U盘之后,界面会出现移动设备插入的提示,选中Open in File Manager 然后点击OK就可以了

找到深度学习的项目代码,然后跟Windows一样的操作,复制粘贴到桌面上。

复制到桌面上之后可以看到,复制到系统上的项目图片,和移动设备的图标。

4.在Tonny中打开项目文件


在Tonny中点击View,然后选择Files,显示文件管理窗口。

在File窗口中点击蓝色的路径就可以一直找到项目文件了,通过点击蓝色的路径中的文件夹的名称蓝色路径下的空白部分就会显示当前路径下的内容。

5.配置Tonny解释器


详细教程见下面文章的第四部分。

树莓派5环境配置笔记 新建虚拟python环境---安装第三方库---配置Thonny解释器:https://chen-hao.blog.csdn.net/article/details/141256300

选则完成之后界面中的路径如下。

6.运行训练模型的程序


然后打开左边的train.py文件,直接点击右上角绿色运行训练程序就行 ,虽然慢点但是是可以跑通的这里我用的是LeNet跑的,我下面的截图是我运行完的结果,训练曲线的弹窗也出来了,OK成功!

报错笔记---安装时报错 ERROR: THESE PACKAGES DO NOT MATCH THE HASHES

上网搜了很多方案 比如加upgrade还有 --no-cache-dir 等等的方法,都不好使,最后的解决方案是在安装pytroch之间先使用一个sudo apt update更新软件包索引包就ok了。

cpp 复制代码
sudo apt update 

2024.8.24 文章入选全站综合热榜感谢认可


结束


这样的话基本上在单片机上进行一些基本的模型推理的基础技术部分就结束了,这样就可以把传感器的数据和深度学习模型直接结合起来,完成一个非常好的实践方式后续我再想想还需不需要尝试一下Docker之类的在树莓派上。

相关推荐
具身AGI3 小时前
物理AI 空间理解:空间物理 信息的三条注入路线
人工智能·深度学习
AOI小白新手上路3 小时前
anomalib 缺陷检测复现笔记:从跑通库到 EfficientAD 落地
人工智能·笔记·机器学习
xxwl5853 小时前
RabbitMQ 学习笔记
笔记·学习·rabbitmq
黑妹天下第一乖3 小时前
第 08 讲:阿加犀 AidGenSE 端侧大模型服务化与 OpenAI 兼容接口
人工智能·嵌入式硬件·深度学习·机器人·iot
加油努力有饭吃4 小时前
DLMS/COSEM 蓝皮书解读(二十六):Auto connect(class_id = 29)—— 自动拨号与主动连接(出站管理)
笔记
2601_962177304 小时前
2026年AI API Gateway怎么选?我整理了6种方案的费用、稳定性和适用场景
网络·人工智能·深度学习·gateway
CarIise4 小时前
Java实训阶段查漏补缺复习笔记1
java·开发语言·笔记
陈卫军老师4 小时前
陈卫军谈商业底线:营销是术,产品是道
经验分享·笔记·流量运营
for_ever_love__5 小时前
PyTorch 张量与 autograd——自动求导怎么工作
pytorch·python·深度学习·自动求导
智者知已应修善业5 小时前
【51单片机只有一个8*8点阵数据如何滚动起来一片595+一组IO+花样】2022-1-15
c++·经验分享·笔记·单片机·蓝桥杯·硬件架构