Jetson Nano安装Numba

Jetson Nano安装Numba

  • 前言
  • 安装步骤
    • 环境依赖
    • 安装
  • AnaConda里安装
    • 安装步骤
  • 以下是坑 (泪目)
    • 不算坑的坑
    • TBB 版本太老
    • LLVM版本太低
  • 最后

前言

近期在Jetson nano上试跑一个姿势估计模型(Pose Estimation Model),后处理时需要用numba。我在nano上安装numba的过程中踩了很多坑…总结一下踩过的坑和解决方式,以及最后成功的安装步骤。
刚接触ML和Jetson nano不久,是个小白,如果下面有什么错误请各位大佬多多指教!!QAQ跪谢。

安装步骤

先直入主题,环境依赖和安装。

环境依赖

  1. 安装setuptools,numpy依赖。
    pip install setuptools
    pip install numpy
  2. 安装llvm依赖,图省事可以直接跟着链接里步骤安装。或者参考他们GitHub自行配置安装。
  3. 安装TBB(Threading Building Blocks),numba内部用来做并行执行的一个库。
# git clone https://github.com/wjakob/tbb.git
# cd tbb/build
# cmake ..
# make -j4
# sudo make install

我开始想图方便,用 sudo apt install libtbb-dev直接进行安装。但是不知道为什么失败了,不过大伙也可以试试。毕竟就一句,失败了也没啥大事,成功了还省事。

  1. 安装llvm lite,用pip安装就可以了:pip install llvmlite=0.xx.0
    下图是版本适配表,大家根据需要自行填入。
    Jetson Nano安装Numba_第1张图片

安装

  1. pip install numba
  2. Terminal试一下import numba,没有报错。成功~
    Jetson Nano安装Numba_第2张图片

AnaConda里安装

还有一种办法是装个AnaConda,里面直接安装numba。个人觉得这个比较方便,毕竟容错率高,不会弄坏系统环境。

我刚开始没找到Jetson nano上用的AnaConda,所有包都sudo pip install/uninstall,sudo apt-get xxx,就老搞坏系统环境,只能所有东西都重装一遍,于是又划过去一天(不是。

安装步骤

  1. 安装conda4aarch64。因为Jetson nano是建在ARMv8(AArch64)平台上的,AnaConda官方版本并不支持,所以需要另外下载安装。
  2. 配置conda,把下面俩频道(channels)加进去:
$ conda config --add channels c4aarch64
$ conda config --add channels conda-forge
  1. 安装numba:conda install -c numba numba

以下是坑 (泪目)

不算坑的坑

在安装numba的时候,我试图跟官网的步骤从源码安装:

git clone git://github.com/numba/numba.git
python setup.py install

然后报错和llvmlite版本不兼容。最后索性直接pip install numba,安装成功。

TBB 版本太老

这个就是pip install numba的时候会跳出来的一个error。参照环境依赖里第三步解决。
Jetson Nano安装Numba_第3张图片

LLVM版本太低

原本是跟着Install-Packages-Jetson-ARM-Family
下载安装numba的,然后没想到它是llvm-7.0.1.src.tar.xz - llvm7.0.1版本,后面也报了一堆错,才发现原来是llvm版本太低不兼容了。

最后

其实中途还有很多没记录下来的坑,但我觉得按上面步骤过一遍的话,那些坑应该都不会踩到,所以就不复现了(也没什么价值

接下来可能写一下怎么注释(annotate)自己的姿势估计模型训练集?如果还能记起来的话……

你可能感兴趣的:(人工智能,nvidia,深度学习)