在RaspberryPi上编译PyTorch

在 RaspberryPi 上编译 TensorFlow 之后，继续用 PyTorch 摧残 RaspberryPi。

安装依赖

sudo apt-get install libopenblas-dev cython3 libatlas-dev m4 libblas-dev

export NO_CUDA=1 # 反正不支持 Pi 上的 GPU
export NO_DISTRIBUTED=1 # 不需要分布式。能消除一个编译错误

git clone --recursive https://github.com/pytorch/pytorch

然后切换到一个已经发行的分支上（例如 v0.3.0）

git checkout -b v0.3.0

编译：

python3 setup.py build

直接安装：

sudo python3 setup.py install

生成whl包（去dist文件夹下收割结果）：

python3 setup.py bdist_wheel

import torch
a = torch.FloatTensor((2,2))
a.add_(3)
print (a)

好像PyTorch可以使用NNPack来提速的，但是不会配置。

自己编译好的都放在了 Google 云端硬盘上。戳这里查看并下载。