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
。
- device_count (int,可选) - 设备数目。默认值为
返回: fluid.CUDAPinnedPlace
对象列表。
返回类型:list[fluid.CUDAPinnedPlace]
代码示例
- import paddle.fluid as fluid
- cuda_pinned_places_cpu_num = fluid.cuda_pinned_places()
- # 或者
- cuda_pinned_places = fluid.cuda_pinned_places(1)