在树莓派5上使用pytroch进行模型训练

本文主要是介绍在树莓派5上使用pytroch进行模型训练—全流程笔记，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

在树莓派上运行pytroch模型🚀

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

在这里插入图片描述

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

文章目录

在树莓派上运行pytroch模型🚀
1.在树莓派上新建一个python的虚拟环境并安装pytroch(CPU)
1+.安装matplotlib绘图库（之后绘制损失曲线用的到）
2.使用的训练代码（仓库/详解博客）
3.通过U盘传输模型代码
4.在Tonny中打开项目文件
5.配置Tonny解释器
6.运行训练模型的程序
报错笔记—安装时报错 ERROR: THESE PACKAGES DO NOT MATCH THE HASHES
结束

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

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

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

新建虚拟环境

python3 -m venv pytroch

激活虚拟环境

source pytroch/bin/activate

在这里插入图片描述

配置清华园镜像

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

在这里插入图片描述

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

sudo apt update

在这里插入图片描述

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

pip3 install torch torchvision torchaudio

在这里插入图片描述
安装成功之后看下pytorch的版本。
启动python

python

导入torch

import torch

查看pytroch版本

troch.__version__

退出python解释器

exit()

在这里插入图片描述

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

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

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一样的操作，复制粘贴到桌面上。
复制到桌面上之后可以看到，复制到系统上的项目图片，和移动设备的图标。
在这里插入图片描述