Python API
is_cuda_available
cuda设备是否在当前机器可用。
get_device_count
获取当前系统中的设备数量。
get_default_device
查询默认的计算结点。
set_default_device
设置默认的计算节点。
get_mem_status_bytes
以字节为单位获取计算设备上的总内存和可用内存。
set_prealloc_config
对原始设备的内存分配器指定预先分配的方案。