MXNet 编译安装 – IYATT-yx 的博客

阅读量： 144

最近更新于 2024-05-05 14:19

测试环境

Android 11 – Termux 0.118 – Ubuntu 20.04 arm64

Ubuntu 20.04 x86_64

Table of Contents

依赖

sudo apt install -y ccache cmake build-essential git libatlas-base-dev graphviz python3 python3-pip

OpenCV：https://blog.iyatt.com/?p=1243

获取源码

目前最新稳定版 1.9.0 – 2022.1.21

git clone --recursive  https://github.com/apache/incubator-mxnet.git --branch=1.9.0

可选扩展

GPU （NVIDIA）加速：https://blog.iyatt.com/?p=2224

编译

cd incubator-mxnet && mkdir build && cd build

# 无 GPU 支持
设置 -DUSE_CUDA=OFF -DUSE_CUDNN=OFF，删掉 -DCMAKE_CUDA_COMPILER

# GPU 支持
cmake -DUSE_CUDA=ON -DUSE_CUDNN=ON -DCMAKE_CUDA_COMPILER=/usr/local/cuda/bin/nvcc -DUSE_CPP_PACKAGE=ON -DBUILD_CPP_EXAMPLES=OFF -DUSE_BLAS=atlas -DCMAKE_BUILD_TYPE=release ..

make -j$(nproc)

sudo make install  # C

cd ../python

python3 -m pip install .  # Python - Python 是调用 C 库，安装 Python 版必须先安装 C 版

可能遇到的错误

我在编译时遇到了找不到 python 的错误，Python3 是确定已经安装的，在 Linux 中目前调用 Python3 要明指明 3，而 python 代表的是 Python2（未安装）。因此我的解决方案是为 Python3 创建一个名为 python 的软链接。

sudo ln -s `which python3` /usr/bin/python

然后再次执行 make 编译

MXNet 编译安装