树莓派3b+安装pytorch

使用Python3 (使用Python2的同学将下面的Python3换成Python,pip3换成pip即可),我的sd卡32G,安装的是pytorch1.0.1。
利用pip search torch可以查看可安装在树莓派上的torch版本。

1.对系统进行更新
sudo apt update && sudo apt upgrade

sudo apt install vim

2.打开交换区设置文件
sudo nano /etc/dphys-swapfile
接着找出常量CONF_SWAPSEIZE,CONF_MAXSWAP,把后面的默认的100改成2048,增加交换区的大小到2G。
然后执行以下命令激活交换文件系统:
sudo /etc/init.d/dphys-swapfile stop

sudo /etc/init.d/dphys-swapfile start
3.接着安装依赖包
sudo apt-get install libopenblas-dev cython3 libatlas-dev m4 libblas-dev cmake
用git下载pytorch源代码:
git clone --recursive https://github.com/pytorch/pytorch
打开pytorch安装目录:
cd pytorch
切换到我们想安装的版本,例如v1.0.1:
git checkout tags/v1.0.1 -b build

git submodule update --init --recursive
4.开始编译pytorch。
编译pytorch之前,需要安装依赖的python包。安装依赖包:
pip install -r requirements.txt

以下4.1和4.2任选一个
4.1 这一步非常重要,如果环境变量设置不正确,编译的过程会报错。例如:
Failed to run 'bash tools/build_pytorch_libs.sh --use-cuda --use-nnpack --use mkldnn --use qnnpack caffe2'
我们可以利用export命令,设置所需要的环境变量,例如我们把cuda支持去掉,因为树莓派不支持cuda。
export NO_CUDA=1

export NO_DISTRIBUTED=1

export NO_MKLDNN=1

export NO_NNPACK=1

export NO_QNNPACK=1
接着我们就可以利用命令:
python3 setup.py build
(不能用sudo python3 setup.py build,否则设置的环境变量不起作用)
对pytorch进行编译,这个过程需要需要耗费几个小时。
当编译完成后,我们就可以利用命令进行安装了
python3 setup.py install
这样pytorch就完成安装了。

4.2或者直接用
sudo -E USE_MKLDNN=0 USE_QNNPACK=0 USE_NNPACK=0 USE_DISTRIBUTED=0 BUILD_TEST=0 python3 setup.py install
若出现error:
Failed to run 'bash ../tools/build_pytorch_libs.sh caffe2'
则在pytorch文件夹下
sudo git submodule update --init

sudo -E USE_MKLDNN=0 USE_QNNPACK=0 USE_NNPACK=0 USE_DISTRIBUTED=0 ./scripts/build_raspbian.sh

使用python3 setup.py build时出现
this caffe2 build also takes long time make: *** [Makefile:144:all] 错误 2相关的错误可以直接放弃build,直接利用4.2的命令
4.3可以通过export MAX_JOBS=2,防止树莓派内存过小,编译卡死的情况,使得编译速度加快。

5.测试pytorch
在python环境中测试例如
import torch a = torch.rand(5,3) print(a)

参考:https://blog.csdn.net/zhouxzh123/article/details/88652312
https://github.com/pytorch/pytorch/issues/26455
https://blog.techforce.top/165.html
https://blog.csdn.net/chengyq116/article/details/83817726

你可能感兴趣的:(树莓派,pytorch,python,pytorch,caffe)