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

最新推荐文章于 2024-08-22 22:17:48 发布

Irving.Gao

最新推荐文章于 2024-08-22 22:17:48 发布

阅读量3.2k

点赞数 1

分类专栏：嵌入式电脑 | 树莓派和Jetson

本文链接：https://blog.csdn.net/qq_45779334/article/details/108640247

版权

嵌入式电脑 | 树莓派和Jetson 专栏收录该内容

45 篇文章 22 订阅

订阅专栏

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

文章目录

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版本，所以需要注意该文章中的软件版本问题。