WRAPPER.md

pyNVML wrapper

Following APIs are supported on OS X, all pyNVML APIs pynvml.nvml* are supported on Linux and Windows.

Initialization

nvmlInit

Initialize nvml.

nvmlShutdown

Shutdown nvml.

System Queries

nvmlDeviceGetCount

Get CUDA device count.

nvmlDeviceGetHandleByIndex

Get CUDA device handle by index.

nvmlDeviceGetIndex

Get CUDA device index.

Device Queries

nvmlDeviceGetMemoryInfo

Get CUDA device memory information.

nvmlDeviceGetName

Get CUDA device name.