#PyCUDA

【GPU加速系列】PyCUDA(一):上手简单操作

PyCUDA可以通过Python访问NVIDIA的CUDA并行计算API。具体介绍和安装可以参考 PyCUDA官网文档和 pycudaPyPI。本文涵盖的内容有:通过PyCUDA查询GPU信息。NumPyarray和gpuarray之间的相互转换。使用gpuarray进行基本的运算。使用Elemen...

pycuda-一些计算函数

单通自定义表达式评估评估GPUArray实例上涉及的表达式可能有些低效,因为为每个中间结果创建了一个新的临时表。模块pycuda.elementwise 中的功能包含有助于生成内核的工具,这些内核在一次传递中评估一个或多个操作数上的多阶段表达式。class pycuda.elementwise.El...
开发笔记 ·2020-03-17