编程|胡渊鸣:import一个“太极”库 让Python代码提速100倍!( 三 )


编程|胡渊鸣:import一个“太极”库 让Python代码提速100倍!
文章图片

有趣的是 , 胡渊鸣介绍 , 即使V的初始浓度是随机设置的 , 但每次都可以得到相似的结果 。
而且和只能达到30fps左右的Numba实现比起来 , Taichi实现由于可以选择GPU作为后端 , 轻松超过了 300fps 。
pip install即可安装
看完上面三个例子 , 你这下相信了吧?
其实 , Taichi就是一个嵌入在Python中的DSL(动态脚本语言) , 它通过自己的编译器将被 @ti.kernel 装饰的函数编译到各种硬件上 , 包括CPU和GPU , 然后进行高性能计算 。
有了它 , 你无需再羡慕C++/CUDA的性能 。
正如其名 , Taichi就出自太极图形胡渊鸣的团队 , 现在你只需要用pip install就能安装这个库 , 并与其他Python库进行交互 , 包括NumPy、Matplotlib和PyTorch等等 。
当然 , Taichi用起来和这些库以及其他加速方法有什么差别 , 胡渊鸣也给出了详细的优缺点对比 , 感兴趣的朋友可以戳下面的链接详细查看:
https://docs.taichi-lang.org/blog/accelerate-python-code-100x
【编程|胡渊鸣:import一个“太极”库 让Python代码提速100倍!】编程|胡渊鸣:import一个“太极”库 让Python代码提速100倍!
文章图片