本文详细介绍了如何在ARM64架构的设备上编译PyTorch，包括设置编译环境、下载依赖、编译PyTorch及torchvision源码。特别提到了在Python3.5中遇到的弱引用错误及其解决方案，以及如何生成whl包供后续安装使用。此外，还讨论了在内存有限的嵌入式设备上进行编译时的注意事项，如创建额外的交换空间来缓解内存限制。
PyTorch源码编译步骤：
1、源码编译环境：
操作系统：debian 9.12
交换空间：1G
Python版本：3.5
硬件：
CPU：RK3399(aarch64)
内存：4G
2、下载依赖包：
下载pytorch及其依赖包时，默认从github上下载，如果网络不好、容易断开时，可在gitee上找到对应包克隆链接，然后修改对应配置文件，进行下载。需要细致耐心。
3、编译主要参数：
设置最大作业数：
export MAX_JOBS=3
pytorch，无cude、无MKL：
export NO_CUDA=1 export NO_DISTRIBUTED=1 export NO_MKLDNN=1 export NO_NNPACK=1 export NO_QNNPACK=1
torchvision源码编译步骤：
1、安装环境依赖：
sudo apt-get install libjpeg-dev zlib1g-dev
2、下载源代码：
git clone -b v0.5.0 https://github.com/pytorch/vision.git torchvision
或者：git clone -b v0.5.0 https://gitee.com/mirrors/vision.git torchvision
3、源码编译：
cd torch
最低0.47元/天 解锁文章
3196
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
