跳转到主要内容

Nvidia CUDA的Python封装

项目描述

Gitlab Build Status https://badge.fury.io/py/pycuda.png Zenodo DOI for latest release

PyCUDA允许您从Python访问Nvidia的CUDA并行计算API。已经存在几个CUDA API的封装,那么PyCUDA有什么特别之处呢?

  • 与对象生命期相关联的对象清理。这种常在C++中称为RAII的惯用方法使得编写正确、无泄漏和崩溃的代码变得容易得多。PyCUDA也了解依赖关系,因此(例如)它不会在分配的内存全部释放之前断开与上下文的关系。

  • 便利性。类似于pycuda.driver.SourceModule和pycuda.gpuarray.GPUArray的抽象使得CUDA编程比使用Nvidia基于C的运行时更加方便。

  • 完整性。如果您愿意,PyCUDA将CUDA的驱动API的全部功能置于您的掌握之中。它还包括与OpenGL互操作性的代码。

  • 自动错误检查。所有CUDA错误都自动转换为Python异常。

  • 速度。PyCUDA的基础层是用C++编写的,所以所有上面的优点几乎都是免费的。

  • 有用的文档

相关地,PyCUDA的姊妹项目PyOpenCLOpenCL提供了类似的、志同道合的计算机功能。

项目详情


版本历史 发布通知 | RSS源

下载文件

下载适用于您平台的文件。如果您不确定选择哪个,请了解更多关于 安装包 的信息。

源分布

pycuda-2024.1.2.tar.gz (1.7 MB 查看哈希值)

上传时间 源代码

支持者

AWS AWS 云计算和安全赞助商 Datadog Datadog 监控 Fastly Fastly CDN Google Google 下载分析 Microsoft Microsoft PSF 赞助商 Pingdom Pingdom 监控 Sentry Sentry 错误记录 StatusPage StatusPage 状态页面