前言
上回我们安装了OpenCV 4.4，相信对源码编译库文件有了一定的了解，这篇我们进一步在树莓派上编译并安装 Pytorch 的最新版本。
PyTorch 1.6 的新特性
PyTorch 1.6 版本增加了许多新的 API、用于性能改进和性能分析的工具、以及对基于分布式数据并行(Distributed Data Parallel, DDP)和基于远程过程调用(Remote Procedure Call, RPC)的分布式训练的重大更新。部分更新亮点包括：
i. 原生支持自动混合精度训练(AMP, automatic mixed-precision training)，只需增加几行新代码就可以提高大型模型训练50-60% 的速度。
i. 为 tensor-aware 增加对 TensorPipe 的原生支持
i. 在前端 API 增加了对 complex tensor 的支持
i. 新的分析工具提供了张量级的内存消耗信息
i. 针对分布式数据并行训练和远程过程调用的多项改进和新功能
增加交换内存（可选）
编译 torch 需要花费大量的内存，在低于 2g 或以下内存的树莓派上，可以通过增加虚拟内存来防止OOM，4g 或 8g 的版本的树莓派可跳过这步。
1 修改配置文件
sudo nano /etc/dphys-swapfil
设置 4g 的交换内存，文件内容如下：
# /etc/dphys-swapfile - user settings for dphys-swapfile package
保存退出，重启服务生效。
sudo service dphys-swapfile restart
查看一下 swap 是否已调整。
swapon -s
PyTorch 安装环境依赖 1 安装依赖首先安装一些编译需要的依赖库：
sudo apt-get install libopenblas-dev cython3 libatlas-base-dev m4 libblas-dev cmake
2 切换虚拟环境
deactivate # 退出之前 OpenCV 的虚拟环境
**编译安装 PyTorch ** 1 设置配置项
export NO_CUDA=1
2 安装库文件
pip3 install numpy pyyaml
✎ Tip
务必确认一下虚拟环境下，已经安装了numpy。没有numpy的话也能成功编译，但是编译出来的PyTorch ****不支持numpy****。PyTorch was compiled without NumPy support。
3 下载源码及支持库
git clone https://github.com/pytorch/pytorch.git
4 生成whl安装包
python3 setup.py bdist_wheel
顺便安装一个CPU 温度和使用率工具s-tui，来监测一下系统状态。
sudo pip install s-tui --ignore-installed
持续满负荷状态：
5 安装 PyTorch
cd dist
看到如下信息，就代表安装成功了。
**编译安装 Torchvision ** 1 下载源码
git clone https://github.com/pytorch/vision.git
2 选择对应版本
pytorch 1.6 对应的 torchvision 是 0.7 的版本，checkout 出来，并安装 PIL 支持。
pip3 install pillow
✎ Tip
编译如遇到以上错误信息，是由于源码中有两处****变量类型错误****，需要用 size_t 强制类型转换一下。修改对应的 seekable_buffer.cpp 和 util.cpp 文件即可。
3 安装 TorchVision
cd dist
搞定！
**运行 yolo v5 ** 1 克隆 yolov5 源码
git clone https://github.com/ultralytics/yolov5
2 软链接到 OpenCV
cd ~/my_envs/pytorch/lib/python3.7/site-packages
✎ Tip
若要删除软链接，用 rm -rf ./cv2 即可，要注意的是千万别在最后添加 /。
3 安装依赖库
pip install tqdm
4 图像推理
测试用最小的模型 yolov5s 对两张图片进行目标检测，识别率还不错，但速度一般，一张 3.8 秒，一张 2.8 秒，大约 0.3fps，后续我们可以对比一下openvino 加速的效果。
cd yolov5
到这里，树莓派里的 pytorch1.6 已经可以正常工作了。
资料下载
若想跳过冗长的编译过程，可以直接下载whl，然后用 pip install 进行安装即可。基于 python 3.7 的版本，除了 pytorch 1.6 + torchvision 0.7，我还编译了最新的 pytorch 1.7 + torchvision 0.8（安装时要注意版本匹配）。本期相关文件资料，可在公众号后台回复：“rpi04”，获取下载链接。
下一篇
我们将开始安装 Tensorflow 的开发环境，
并运行一下 tensorflow lite，
看一下裸板树莓派推理的极限速度，
敬请期待...
