在 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 云端硬盘上。戳这里查看并下载

参考资料

留下评论