树莓派上安装 PyTorch

发表于 |

买不起 jetson nano，难过……

前言

  Raspbian 源上的 PyTotch 较老，要用 1.0 以上的库必须自己编译才行。然而 arm 版的 PyTorch 有很多不支持，需要比较复杂的配置才行。所以记录一下我的安装过程。

增加 SWAP

1
sudo nano /etc/dphys-swapfile

修改 CONF_SWAPSIZE 和 CONF_MAXSWAP，把后面的数字改为 2048 以上（2GB），然后使 SWAP 空间生效：

1
2
sudo /etc/init.d/dphys-swapfile stop
sudo /etc/init.d/dphys-swapfile start

通过下面命令确认：

1
free -h

依赖

1
sudo apt install libopenblas-dev libblas-dev libatlas-dev m4 cmake cython3 python3-dev python3-yaml python3-cffi python3-setuptools

1
pip install numpy pyyaml

获取源码

1
2
3
4
git clone https://github.com/pytorch/pytorch.git
cd pytorch
git checkout v1.2.0
git submodule update --init --recursive

1.2.0使用的protobuf有一个Bug会导致报undefined reference to `__atomic_fetch_add_8’的错误，参考Pytorch Issue，升级protobuf到最新即可解决：

1
git submodule update --remote third_party/protobuf

设置参数

1
2
3
4
5
6
export NO_CUDA=1
export NO_DISTRIBUTED=1
export NO_MKLDNN=1
export NO_NNPACK=1
export NO_TEST=1
export MAX_JOBS=2 #编译时最大CPU核心数

编译

1
python3 setup.py build

我的树莓派3B+ 编译了40多个小时 😭，我看的所有教程都说只需要 7~8个小时。

编译时可能会出现 subprocess.CalledProcessError: Command '['cmake', '--build', '.', '--target', 'install', '--config', 'Release', '--', '-j', '4']' returned non-zero exit status 2. ，这时要检查一下前面的东西。如果反复出现，考虑一下用 Python3.6 而不是 3.7（buster 默认是 3.7）

安装

1
sudo -E python3 setup.py install

测试

1
import torch

__C.so 不存在

去到 torch 安装的文件夹（我的是 /usr/local/lib/python3.7/dist-packages/torch），将 __C 开头那个文件重命名为 __C.so，同时将 __dl 开头那个文件重命名为 __dl.so

__C 未定义

去 torch 的github文件夹，里面有个 test 文件，在里面运行 python，再试试 import torch

参考