Jetson Nano——Ubuntu16.04安装numba并利用numba实现python代码加速

因为搜遍了全网都没有可以能够直接安装成功的教程，博主花了半天成功安装，此处采坑记录，和大家作以分享。

CUDA导入测试

首先，查看cuda的bin目录下是否有nvcc：

cd /usr/local/cuda/bin

ls

如果存在，直接将cuda路径加入到系统路径即可，即复制在最后即可：(需要注意的是，在此处添加的路径要和自己配置文件的路径一致)

sudo nano ~/.bashrc

export CUBA_HOME=/usr/local/cuda
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda/bin:$PATH

然后更新配置文件：

source ~/.bashrc

再次执行nvcc -V就可以看到相应cuda版本了，如下：

安装numba

numba可以基于llvm动态生成优化代码，提高python的执行效率，使用很方便，只需要给python代码加上修饰器就好了。由于numba依赖llvm，需要在ubuntu上事先安装好llvm。

安装LLVM依赖

LLVM官网

建议安装10.0.0版本的Pre-Built Binaries版本：
选择AArch64-Linux版本下载即可，如果下载速度过慢，参考waiwang资源下载——使用迅雷加速资源下载。

下载好后：

• 解压压缩包

tar xJvf clang+llvm-10.0.0-aarch64-linux-gnu.tar.xz

• 修改环境变量

sudo nano ~/.bashrc

• 将下面环境变量path放入文件最后一行即可

export PATH=$PATH:/home/nano/clang+llvm-10.0.0-aarch64-linux-gnu/bin

• 导入环境变量

source ~/.bashrc

• 测试版本

clang++ -v   #测试是否安装成功

安装Cython、llvmlite、numpy、setuptools依赖库

pip3 install cython
pip3 install llvmlite
pip3 install numpy
pip3 install setuptools

安装numba

直接pip3 install numba会build报错，规定版本号后，直接成功：

pip3 install numba==0.31

毫无错误，直接安装成功！

可借鉴安装教程（依然有需要修正的地方）：
Numba Ubuntu包安装指南(含llvm、llvmlite，最简单)
需要注意的是，该博客中的LLVM下载版本型号有误，按照我上面说的版本下载，应当下载的LLVM版本应为LLVM 10.0.x or 9.0.x。Jetson nano需要下载的所有软件版本都是Linux-aarch64版本，所以需要注意该文章中的软件版本问题。

报错解决

• 1.安装完LLVM7.0.1后pip3 install llvmlite报错：

• 解决方法：下载的LLVM版本应为LLVM 10.0.x or 9.0.x

• 2.pip3 install numba报错：
  
  经过检查后，发现问题在于没有安装Cython库：
  

• 解决方法：下载的Cython库

pip3 install cython

参考文章：
超详细Python Cuda零基础入门教程
ubuntu16.04安装numba
Jetson Nano安装Numba遇到的一些坑