1. 依赖环境

1.0 docker镜像拉取创建容器

$ sudo docker pull libtorch-cpu:1.0
$ sudo docker run -it -d -v ~/Desktop:/workspace -p 9009:22 --name libtorch libtorch-cpu:1.0
$ sudo docker exec -it libtorch bash

1.1 安装gcc compiler

$ apt update
$ apt install build-essential

1.2 安装python3.7

$ apt install python3.7

1.3 安装cmake3.20.5

$ apt install ninja-build
$ apt install cmake

1.4 确定处理器个数(此步骤可以省略)

$ CPU_NUM=$(grep processor /proc/cpuinfo | awk '{field=$NF};END{print field+1}')
$ echo "CPU number = "$CPU_NUM

2. 下载源码

$ git clone --recursive <https://github.com/pytorch/pytorch.git>
$ cd pytorch
$ git checkout v1.7.1
$ git submodule sync
$ git submodule update --init --recursive

$ export LIBTORCH_ROOT=$(pwd)

3. aarch64-linux-gnu-gcc编译源码

cmake .. -G"Unix Makefiles" \\\\
-DCMAKE_CXX_COMPILER=/marvell-tools-265.0/bin/aarch64-marvell-linux-gnu-g++ \\\\
-DCMAKE_BUILD_TYPE=Release \\\\
-DCMAKE_INSTALL_PREFIX=/opt/libtorch/ \\\\
-DPYTHON_EXECUTABLE=$(which python3)\\\\
-DBUILDING_WITH_TORCH_LIBS=ON \\\\
-DATEN_THREADING=NATIVE -DUSE_OPENCV=OFF \\\\
-DBUILD_EXAMPLES=ON \\\\
-DUSE_CUDA=OFF \\\\
-DUSE_FBGEMM=OFF \\\\
-DHAVE_STD_REGEX=0 \\\\
-DHAVE_POSIX_REGEX=0 \\\\
-DHAVE_STEADY_CLOCK=0

4. 测试