nvidia cuda api的python绑定。
pynvx的Python项目详细描述
Pynvx
Python bindings for NVIDIA CUDA APIs.
原料药
初始化
cudainit
cudainit()->;整数
>;初始化CUDA运行时。
系统查询
cudasystemgetdriverversion
cudasystemgetdriverversion()->;int
>;获取CUDA驱动程序版本。
cudasystemgetruntimeversion
cudasystemgetruntimeversion()->;int
>;获取CUDA运行时版本。
cudadevicegetcount
cudDeviceGetCount()->;整数
>;获取CUDA设备计数。
cudadevicegethandles
cudadevicegethandles()->;列表[int]
>;在列表中获取CUDA设备句柄。
设备查询
cudagetname
cudagetname(句柄)->;str
>;获取CUDA设备名称。
cudagetclockrate
CUDAGetClockRate(手柄)->;积分
>;以千赫兹为单位获取CUDA设备时钟速率。
cudagetcomputecapability
cudagetcomputecapability(句柄)->;列表[int]
>;在列表[major,minor]中获取CUDA修订号,定义设备的计算能力。
cudagetmajor
cudagetmajor(手柄)->;int
>;获取定义设备计算能力的CUDA主要修订号。
cudagetminor
cudagetminor(句柄)->;int
>;获取定义设备计算能力的CUDA次要修订号。
cudagetmeminfo
cudagetmeminfo(句柄)->;列表[整数]
>;获取列表中设备的CUDA内存信息[总计,可用,已用]。
cudagetmemtotal
cudagetmemtotal(句柄)->;int
>;获取设备的CUDA总内存。
cudagetmemfree
cudagetmemfree(句柄)->;int
>;获取设备的CUDA可用内存。
cudagetmemused
cudagetmemused(句柄)->;int
>;获取设备的CUDA已用内存。
cudagetmultiprocessorcount
cudagetmultiprocessorcount(handle)->;int
>;获取设备上的CUDA多处理器编号。
cudagetpcibusid
cudagetpcibusid(句柄)->;int
>;获取设备的CUDA PCI总线标识符。
cudagetpcideviceid
cudagetpcideviceid(句柄)->;int
>;获取设备的CUDA PCI设备(有时称为插槽)标识符。
cudagetpcidomainid
cudagetpcidomainid(句柄)->;int
>;获取设备的CUDA PCI域标识符。