当前位置:网站首页 > PyTorch框架 > 正文

pytorch模型部署到树莓派(pytorch 模型部署)



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


在这里插入图片描述


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




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

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

新建虚拟环境

 
      

激活虚拟环境

 
      

在这里插入图片描述

配置清华园镜像

 
      

在这里插入图片描述

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

 
      

在这里插入图片描述

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

 
      

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

 
      

导入torch

 
      

查看pytroch版本

 
      

退出python解释器

 
      

在这里插入图片描述


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

 
       

进入到环境中之后直接安装就行
在这里插入图片描述


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

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

在这里插入图片描述

博客地址:https://chen-hao.blog.csdn.net/article/details/
在这里插入图片描述


将代码下在U盘里然后直接插入,这里为了避免远程传输可能出现的各种风险,先直接用硬件传输代码。
在这里插入图片描述
插入U盘之后,界面会出现移动设备插入的提示,选中Open in File Manager 然后点击OK就可以了
在这里插入图片描述
找到深度学习的项目代码,然后跟Windows一样的操作,复制粘贴到桌面上。在这里插入图片描述
复制到桌面上之后可以看到,复制到系统上的项目图片,和移动设备的图标。
在这里插入图片描述






在Tonny中点击View,然后选择Files,显示文件管理窗口。
在这里插入图片描述
在File窗口中点击蓝色的路径就可以一直找到项目文件了,通过点击蓝色的路径中的文件夹的名称蓝色路径下的空白部分就会显示当前路径下的内容。
在这里插入图片描述



在这里插入图片描述

选则完成之后界面中的路径如下。
在这里插入图片描述


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

在这里插入图片描述

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

 
            

在这里插入图片描述
在这里插入图片描述


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

到此这篇pytorch模型部署到树莓派(pytorch 模型部署)的文章就介绍到这了,更多相关内容请继续浏览下面的相关推荐文章,希望大家都能在编程的领域有一番成就!

版权声明


相关文章:

  • resnet模型训练过程(resnet预训练模型pytorch)2025-06-28 10:36:07
  • pytorch模型部署方案(pytorch模型的保存与加载)2025-06-28 10:36:07
  • 怎么删除虚拟环境中的pytorch(pytorch 虚拟环境)2025-06-28 10:36:07
  • 删除pytorch虚拟环境(pycharm怎么删除虚拟环境)2025-06-28 10:36:07
  • pytorch模型部署(pytorch模型部署到树莓派)2025-06-28 10:36:07
  • pytorch模型部署到安卓(pytorch 安卓)2025-06-28 10:36:07
  • pytorch模型部署onnx(pytorch模型部署到web)2025-06-28 10:36:07
  • pytorch模型部署到树莓派(pytorch搭建模型)2025-06-28 10:36:07
  • pytorch模型部署onnx(pytorch模型部署 django)2025-06-28 10:36:07
  • 服务器配置pytorch环境(pytorch如何在服务器上跑)2025-06-28 10:36:07
  • 全屏图片