cuda_pinned_places

  • paddle.fluid.cuda_pinned_places(device_count=None)[源代码]

该接口创建 device_count 个 fluid.CUDAPinnedPlace 对象，并返回所创建的对象列表。

如果 device_count 为 None，则设备数目将由环境变量 CPU_NUM 确定。如果未设置 CPU_NUM 环境变量，则设备数目会默认设为1，也就是说， CPU_NUM=1。 CPU_NUM 表示在当前任务中使用的设备数目。如果 CPU_NUM 与物理核心数相同，可以加速程序的运行。

  • 参数：
    • device_count (int，可选) - 设备数目。默认值为 None。

返回: fluid.CUDAPinnedPlace 对象列表。

返回类型：list[fluid.CUDAPinnedPlace]

代码示例

  1. import paddle.fluid as fluid
  2. cuda_pinned_places_cpu_num = fluid.cuda_pinned_places()
  3. # 或者
  4. cuda_pinned_places = fluid.cuda_pinned_places(1)