[人工智能]imx6ull运行C++libtorch示例程序

1. 获取32位libtorch的编译版本

• ?GitHub - ljk53/pytorch-rpi: Share PyTorch binaries built for Raspberry Pi?被人编译的应用于树莓派的非官方编译版本。
• 解压获得可用于armv7架构的libtorch，为后续交叉编译提供运行库和头文件

2.在vscode编写example_app.cpp CMakeList.txt 交叉编译arm_linux_setup.cmake文件

英文教程链接https://pytorch.org/cppdocs/installing.html?

中文教程链接https://pytorch.apachecn.org/#/docs/1.7/44

交叉编译设置教程引用https://blog.csdn.net/whahu1989/article/details/104114515

//example_app.cpp
#include <torch/torch.h>
#include <iostream>

int main() {
  torch::Tensor tensor = torch::rand({2, 3});
  std::cout << tensor << std::endl;
}

//CMakeList.txt
cmake_minimum_required(VERSION 2.8)

project(pytorch)

find_package(Torch REQUIRED)

set( SRC_LIST ${PROJECT_SOURCE_DIR}/source/example_app.cpp)

set (EXECUTABLE_OUTPUT_PATH ${PROJECT_SOURCE_DIR}/bin)

find_library(OPENBLAS_LIB NAMES openblas HINTS /home/xxx/openblas/lib)

add_executable(example_app ${SRC_LIST})

target_link_libraries(example_app "${TORCH_LIBRARIES}" ${OPENBLAS_LIB})

set_property(TARGET example_app PROPERTY CXX_STANDARD 14)

//arm_linux_setup.cmake
set(CMAKE_SYSTEM_NAME Linux)

set(CMAKE_SYSTEM_PROCESSOR arm)

set(tools /usr)

set(CMAKE_C_COMPILER ${tools}/bin/arm-linux-gnueabihf-gcc)

set(CMAKE_CXX_COMPILER ${tools}/bin/arm-linux-gnueabihf-g++)

?

?

?3.使用cmake得到makefile文件并编译

mkdir build
cd build
//指定libtorch绝对路径和交叉编译文件
cmake -DCMAKE_PREFIX_PATH=/home/xxx/work_dir/pytorch/libtorch -DCMAKE_TOOLCHAIN_FILE=../arm_linux_setup.cmake .. 

cmake --build . --config Release  //编译

4.编译发现依赖libopenblas库，接着交叉编译该库文件

https://github.com/xianyi/OpenBLAS.git? openblas开源库

下载source code压缩包，并解压，切换到openblas目录，最后开始编译，若交叉编译器没有gfortran文件使用 sudo apt-get install gfortran-arm-linux-gnueabihf 安装

//交叉编译
make HOSTCC=gcc CC=arm-linux-gnueabihf-gcc FC=arm-linux-gnueabihf-gfortran TARGET=ARMV7
make PREFIX=/home/xxx/Openblas install

5.编译成功，拷贝至开发板运行，提示缺少libtorch.so库，查看example_app程序运行依赖的库（readelf -d example_app），并将相应的库拷贝至开发板的usr/lib文件夹（使用sudo命令后记得修改文件权限?sudo chmod 777 libopenblas.so.0 ）另外注意：pc交叉编译器版本不要大于开发板的版本，因为gcc库文件是向下兼容！！！

?6.运行结果

?

?

?

?