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 960MGPU属性
通过device.gpu(id)获取了gpu的引用之后,我们就可以通过device.get_gpu_properties函数来获取属性:
>>> prop = C.device 继续阅读与本文标签相同的文章
上一篇 :
Java并发核心知识讲解
-
Docker安装Gitlab和Gitlab-Runner并实现项目的CICD
2026-05-17栏目: 教程
-
Linux搭建自己的Docker镜像仓库
2026-05-17栏目: 教程
-
MySQL8.0之窗口函数
2026-05-17栏目: 教程
-
推荐一款数据同步工具:FlinkX
2026-05-17栏目: 教程
-
蚂蚁金服自研数据库OceanBase如何登顶TPC-C
2026-05-17栏目: 教程
