配置Pytorch的环境,我自己着实花了很久的时间,遇到了很多的问题,问了让大家少走弯路,我这边写一下一个成功的案例。这边先要感谢一下两位博主的文章。
在 RaspberryPi 上编译 PyTorch
Pytorch on RaspberryPi
前人种树后人乘凉,希望我的文章也能帮助到一些人。 首先树莓派上面是自带了Python2.7的版本的,由于我们这边是要在Python3.5上面配置Pytotch,所以我们先安装一下Python3。
sudo apt install python3
输入python3,安装成功的界面如下:
然后通过pip3 list 查看已经安装的包,可以看见常用的一些包都已经帮我们安装好了(例如numpy,pillow等等)。需要注意的时,这边安装的Python版本是3.5的, 如果需要安装Python3.6.X版本的话,需要自己去官网下载 相关的安装包 并解压编译安装,相对要耗时一些,可能会遇到其他一些奇奇怪怪的问题,这个后续我会再写篇博客讲讲如何安装Python3.6.X并在此版本上配置pytorch。
之前呢,我是按照两位博主的教程,一步一步设置环境变量,然后下载源码,编译安装,我记得大概编译了十几个小时吧,最后还是安装失败了。所以,我这边建议向直接安装成功的小伙伴,可以直接按照我这边的教程( 借助一下两位博主提供的torchXXX.whl文件 ),直接安装。当然了,要是想自己动手尝试了,我也是非诚提倡的,可以遇到很多问题,踩很多坑.....能多了解点东西吧反正。这边torch3.5的版本的话,在这边下载,尊重一下原作者的辛苦成果。
torch-0.4.0a0+3749c58-cp35-cp35m-linux_armv7l.whl
下载完了之后,先cd到存放torchXXX.whl的文件夹,直接安装,命令如下。
实践截图:
具体代码:
pip3 install torch-0.4.0a0+3749c58-cp35-cp35m-linux_armv7l.whl
安装成功截图:
然后打开Python3 ,进行测试
根据报错的问题,我们就查看一下这个libopenblas是个什么东西,是否需要安装或者如何安装这个软件。我们首先利用apt-cache search 命令来查找相应的libopenblas的安装包。接着根据查找到的内容,安装相应的软件。
具体代码如下:
sudo apt-cache search libopenblas
sudo apt install libopenblas-dev
实践截图如下:
等待安装成功之后,然后我们再来测试一下import torch命令。
具体代码如下:
import torch
a = torch.FloatTensor((2,2))
print(a)
a.add(3)
print(a)
实践截图如下:
好的,到此为止,我们就安装好pytorch了。一般情况下,我们需要加载一些图片数据,所以常常需要用到torchvison,所以此处我们就一起安装了torchvison。具体命令如下:
pip3 search torchvision
pip3 install -i https://mirrors.aliyun.com/pypi/simple torchvision
为了以往万一,我们打开python3,导入一下torchvision测试一番:
好的,没有问题!本期博客到此为止。如果觉得本文不错,记得给博主点赞鼓励啊~
待我踩坑归来,再给大家分享踩坑的经验。