树莓派 4B python环境整理 树莓派配置pytorch

admin2024-06-01  19

树莓派3b+配置pytorch和opencv

pytorch

在树莓派安装pytorch并不容易,因为最常用的pip安装方式无法使用,会直接报错(笔者多次报错,且目前没有找到解决方法)。在网上搜寻资料许久,又经过几天的尝试,终于完成了pytorch的配置。

源码编译

相信不少人直接在网上搜索,搜索到直接使用源码编译的方式较多,这里不再赘述。但笔者在使用这一方法时,由于树莓派没有vpn经历了几次下载失败。并且在尝试使用这一行代码时,会出现或因网络原因而停滞不前的情况。

git submodule update --init --recursive

最后尝试使用Windows端挂上梯子,通过git将代码clone,然后将源文件通过ftp传入树莓派进行编译。花了大量时间做好了编译准备,结果却在正式编译一段时间后,树莓派直接死机重启(现在想起来很可能是因为电源的电压电流不够导致的,当时使用的是普通5v1A充电头),编译的文件也损坏了(第二次尝试编译时报错)。

就笔者个人情况而言,并不推荐在树莓派3b+上使用源码编译的方式。

直接使用whl文件安装

笔者经过一番搜索,在git上找到了pytorch专为arm架构使用的whl文件。

https://github.com/nmilosev/pytorch-arm-builds

上面有详细的操作提示,有包括torch1.1.0,1.2.0;torchvision0.3.0,0.4.0在内的多个版本,大家可以根据需要,选择对应的版本自行下载。

另外,感谢,笔者找到了相对更新的1.3.0和0.4.0版本,下载后安装成功。

安装前建议先更换镜像源和pip源,有关教程可自行百度。

安装步骤:

首先安装必要的依赖包

sudo apt-get install libopenblas-dev

安装numpy等依赖库(没有numpy的话也能成功编译,但是编译出来的PyTorch不支持numpy)

pip3 install numpy pyyaml

安装pytorch

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

最后,尝试在python命令行下输入

import torch
print(torch.rand(2,2))

得到结果

tensor([[0.9043, 0.8893],
        [0.5651, 0.9442]])

表明torch已配置成功。

opencv-python的配置

whl文件安装

opencv的配置相对简单,其同样有源码编译及whl文件安装两种方式,这里笔者直接采用了whl的安装方式。

opencv的whl文件来源于https://www.piwheels.org/packages.html

将文件下载完成并传入树莓派后,首先使用如下代码安装依赖包
(注,树莓派不支持opencv4.0)

sudo apt install libqt4-test libqtcore4 libavutil56 libavformat58 libatk1.0-0 libcairo-gobject2 libavcodec58 libgtk-3-0 libpango-1.0-0 libqtgui4 libgdk-pixbuf2.0-0 libtiff5 libwebp6 libopenexr23 libswscale5 libjasper1 libcairo2 libpangocairo-1.0-0 libilmbase23

接着输入

pip3 install  #你下载的版本

即可完成opencv的安装。


本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明原文出处。如若内容造成侵权/违法违规/事实不符,请联系SD编程学习网:675289112@qq.com进行投诉反馈,一经查实,立即删除!