背景

最近要用树莓派部署神经网络相关的任务,因而先在树莓派上安装pytorch实验一番,之后再尝试各种神经网络推理框架。

关于树莓派安装pytorch,我参考了这些文章(油炸花生米34 梦远花落白衣衫 以及dawnbreakerzhouxzh123 )。

首先想到的方法当然是pip直接安装,通过pip搜索torch
在这里插入图片描述
可以看到有一个torch-raspi(0.4.0)版本,针对树莓派3B,ARMv7L指令集架构编译的Linux内核。

比较遗憾的是似乎并不能直接安装新版本的,而是需要编译安装(本该是交叉编译完成的活啊)。

上面四人已经给了编译方法了,在此不再重复。

编译好的文件(分享)

在这里分享一下已经编译好的文件,针对python3.7,各位就免去复杂的编译,直接安装就可以:)
torch-1.3.0a0+de394b6-cp37-cp37m-linux_armv7l.whl 提取码:ioat
torchvision-0.4.1a0+a263704-cp37-cp37m-linux_armv7l.whl 提取码:g0sz

安装Pytorch

安装必要的依赖包sudo apt-get install libopenblas-dev

安装numpy pip3 install numpy pyyaml(没有numpy的话也能成功编译,但是编译出来的PyTorch不支持numpy)

安装torch pip3 install torch-1.3.0a0+de394b6-cp37-cp37m-linux_armv7l.whl

安装torchvision pip3 install torchvision-0.4.1a0+a263704-cp37-cp37m-linux_armv7l.whl

安装界面截图参考:
在这里插入图片描述

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐