numba

本文介绍了Numba这一JIT编译器,它能将Python函数编译为代码,让Python数组运算速度接近C或FORTRAN语言。还阐述了Python慢的原因及Numba换编译器的作用,介绍了装饰器概念,说明了Numba简单、对循环有效等优势,以及导入、传入装饰器编写函数的使用方法。

numba是一款将Python函数编译为代码的JIT编译器,经过numba编译的Python代码(仅限数组运算),其运行速度可以接近C或FORTRAN语言。
Python之所以慢,是因为它是靠CPython编译的,numba的作用是给Python换一种编译器。

装饰器
装饰器本身是一个函数,是函数的函数,目的是增减函数的功能,比如首先定义一个输出当前时间的函数,再定义一个规定时间格式的函数,把后一个函数作用在前一个函数上,就是一个装饰器,作用是用特定格式输出当前时间。

numba的优势

  • 简单,往往只要1行代码就有惊喜。
  • 对循环(loop)有奇效。
  • 兼容常用的科学计算包,如numpy、cmath等。
  • 可以创建ufunc。
  • 会自动调整精度,保证准确性。

使用方法

  • 导入numba及其编译器

    import numba
    from numba import jit
    
  • 传入numba装饰器jit,编写函数

    #传入jit,numba装饰器中的一种
    @jit(nopython=True)
    def function():
        ....
    

参考资料
如何用numba加速python?
numba,让python速度提升百倍

### 如何在不同操作系统上安装numbaNumba 是一个用于加速 Python 代码的 JIT 编译器,支持多种硬件架构和操作系统。以下是针对不同操作系统的安装方法: #### Windows 系统 在 Windows 系统中,推荐使用 `conda` 或 `pip` 安装 Numba。如果系统已经安装了 Anaconda 或 Miniconda,可以通过以下命令安装: ```bash conda install numba ``` 如果更倾向于使用 `pip`,可以运行以下命令[^1]: ```bash pip install numba ``` #### macOS 系统 macOS 用户同样可以选择 `conda` 或 `pip` 进行安装。对于 Conda 用户,可以直接执行: ```bash conda install numba ``` 而对于 Pip 用户,则需要确保系统已安装 LLVM,并通过以下命令安装[^3]: ```bash pip install numba ``` #### Linux 系统(以 Ubuntu 为例) 在 Ubuntu 系统中,推荐先通过系统包管理器安装 LLVM,然后使用 `pip` 安装 Numba。例如,在 Ubuntu 18.04 上可以按照以下步骤操作: ```bash sudo apt update sudo apt install llvm pip3 install llvmlite==0.33.0 pip3 install numba==0.50.0 ``` 此方法适用于大多数基于 Debian 的 Linux 发行版[^3]。 #### 注意事项 - 如果需要使用缓存功能或并行计算,可以参考装饰器选项如 `@njit(cache=True)` 和 `@njit(parallel=True)`[^4]。 - 在 GPU 加速场景下,确保系统已安装 NVIDIA CUDA 工具链或 AMD ROCm 平台[^5]。 ### 示例代码 以下是一个简单的 Numba 使用示例,展示如何利用 `@njit` 装饰器加速函数: ```python from numba import njit @njit def sum_function(a, b): return a + b result = sum_function(10, 20) print(result) ```
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值