CNTK中GPU信息的获取
CNTK中GPU信息的获取
device接口
CNTK提供了device接口,可以访问gpu的几个基本参数。
获取所有的设备
首先可以通过cntk.device.all_devices方法来获取当前的设备
>>> C.device.all_devices()
(GPU[0] GeForce GTX 960M, CPU)
获取GPU
知道了系统里有多少设备了之后,就可以通过设备号来通过device.gpu来访问GPU设备了。
例:
>>> C.device.gpu(0)
GPU[0] GeForce GTX 960M
GPU属性
通过device.gpu(id)获取了gpu的引用之后,我们就可以通过device.get_gpu_properties函数来获取属性:
>>> prop = C.device