树莓派4b搭建PyTorch(for buster)

在树莓派4b上安装pytorch,搭载人工智能学习框架。

1、安装python3.5

1.在buster中,python默认的版本是2.7和3.7,但是由于pytorch还没有针对python3.7的whl包,所以我们选择安装与pytorch最兼容的3.5版本。

sudo apt-get install python3.5 #默认安装的是python3.5.4

2.如果不想使用python2.7,可以将其卸载(最好不要随意卸载3.7,可能造成一些问题)。

sudo apt-get autoremove python2.7

卸载完后,如果想敲python直接出现python3.5的信息,那么:

先删除之前python的链接

sudo rm /usr/bin/python

创建python3.5的链接

sudo ln -s /usr/bin/python3.5 /usr/bin/python   

之后输入python就是python3.5了

如果设置为sudo ln -s /usr/bin/python3.5 /usr/bin/python3 那么调出python3.5就需要输入python3

如果设置python3.7,那么则需要设置为 sudo ln -s /usr/bin/python3.7 /usr/bin/python3

到这里,输入python3.5后,你应该可以看到如下信息:
在这里插入图片描述

3.为python3.5安装pip。
使用:

cd /tmp
wget https://bootstrap.pypa.io/get-pip.py #这里可能需要FQ才能正常访问
sudo python3.5 ./get-pip.py

这样会卸载原来的pip,为python3.5安装好pip。
使用:

pip3.5

ps :网上安装pip方法一大堆,可以从官网下载源文件再进行安装,此处不再赘述。

2、安装pytorch

1、首先我们下载网上某些大神提供的torchXXX.whl文件
下载链接:https://pan.baidu.com/s/1w4su6UTgZLncw40L26YKCA
提取码:re91

2、下载后用FileZilla软件将whl文件导入到/home/pi中(Linux文件系统权限比较麻烦)。
3、cd到pi文件目录
进行手动安装

sudo pip3.5 install torch-0.4.0a0+3749c58-cp35-cp35m-linux_armv7l.whl

显示 sucessfully installed torch-0.40a0a+3749c58 则安装成功

4、安装libopenblas模块

sudo apt-cache search libopenblas
sudo apt install libopenblas-dev

5、卸载原来的numpy模块:

sudo pip3.5 uninstall numpy

下载以下版本的numpy,放到pi文件夹中,进行手动安装
下载链接:https://pan.baidu.com/s/1w4su6UTgZLncw40L26YKCA
提取码:re91

sudo pip3.5 install numpy-1.17.2-cp35-cp35m-linux_armv7l.whl

安装成功后显示 successfully installed numpy-1.17.2

6、安装 libatlas-base-dev

sudo apt-get install libatlas-base-dev

7、修改文件权限。cd到numpy文件目录下

cd /usr/local/lib/python3.5/dist-packages/numpy

授予文件权限

sudo chmod 777 init.py

8、打开__init__.py文件, nano init.py 找到:from . import random

在这里插入图片描述
我们在from . import random前面加#注释掉。 保存,重启树莓派:reboot

9、打开python,引入模块import torch,可以看到成功了:

在这里插入图片描述

注:本文一部分内容来源于网络搜索,通过结合自己的实际操作整理并呈现出来。

发布了31 篇原创文章 · 获赞 2 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/qq_27149279/article/details/105178970