RK3588编译MXNet框架

目录

1. 背景

2.编译MXNet准备

3.开发板编译

---

1. 背景

        MXNet（也称为Apache MXNet或incubator-mxnet）是一个开源的深度学习框架，它最初由华为和亚马逊AWS共同开发，并于2017年成为Apache软件基金会的孵化项目。MXNet旨在提供高效、灵活和可扩展的深度学习框架，可以用于训练和部署机器学习模型。以下是MXNet的一些关键特点和功能：

1. 多语言支持：MXNet支持多种编程语言，包括Python、C++、Scala、Julia和R。这使得MXNet适用于不同背景和编程偏好的开发人员。

2. 灵活性：MXNet提供了高度灵活的符号式和命令式编程接口。符号式编程使您能够定义计算图，以便进行高效的自动微分和优化，而命令式编程更适合快速的原型设计和调试。

3. 分布式训练：MXNet支持分布式训练，可以在多个GPU或多台机器上并行训练深度学习模型，以加速训练过程。它还支持云端和本地集群的分布式计算。

4. 跨平台支持：MXNet可以在各种硬件设备上运行，包括CPU、GPU和专用AI加速器。它还支持多个操作系统，包括Linux、Windows和macOS。

5. 深度学习库集成：MXNet可以与其他深度学习库（如Gluon、Keras、TensorFlow）无缝集成，从而扩展其功能并允许用户使用他们熟悉的工具。

6. 自动微分：MXNet提供了自动微分功能，使您能够轻松计算损失函数的梯度，这对于训练神经网络模型至关重要。

7. 模型部署：MXNet支持将训练好的模型部署到不同的生产环境中，包括云端服务器、移动设备和嵌入式系统。

8. 社区支持：作为一个Apache孵化项目，MXNet拥有活跃的开发社区，提供了文档、教程和许多示例来帮助用户入门和解决问题。

        MXNet是一个功能强大且灵活的深度学习框架，适用于各种深度学习应用，从计算机视觉到自然语言处理。它的跨平台支持、多语言接口和分布式训练功能使其成为一个受欢迎的工具，用于构建和部署深度学习模型。

2.编译MXNet准备

        下载源码需要用到github下载，我这里直接给百度网盘，下载快速一下，链接在这里

        链接: https://pan.baidu.com/s/1mUafX96RAi10OV8VamFeiw 提取码: 1fv3 

3.开发板编译

将mxnet.zip拷贝到RK3588的home目录下

sudo apt update

sudo apt -y install git cmake ninja-build gfortran liblapack* libblas* libopencv* libopenblas* python3-dev python3-pip python-dev virtualenv

sudo apt install  cython

unzip mxnet.zip

cd mxnet

mkdir build 

cd build

        cmake \

        -DUSE_SSE=OFF \

        -DUSE_CUDA=OFF \

        -DUSE_OPENCV=ON \

        -DUSE_OPENMP=ON \

        -DUSE_SIGNAL_HANDLER=ON \

        -DBUILD_CYTHON_MODULES=ON \

        -DCMAKE_BUILD_TYPE=Release \

        -GNinja ..

ninja -j8 # ps:这里需要花费很长的时间

cd ../python

sudo pip3.7 install -e . 
sudo python3.7 setup.py install --user # ps:这里的Python版本根据自己的版本修改

pip3.7 list

最后做推理，可以正常加载模型

# 注意numpy的版本需要1.21.6

pip3.7 install numpy==1.21.6 -i https://pypi.tuna.tsinghua.edu.cn/simple