cuda_pinned_places¶

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

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

如果 device_count 为 None，实际设备数目将由当前任务中使用的GPU设备数决定。用户可通过以下2种方式设置任务可用的GPU设备：

• 设置环境变量 FLAGS_selected_gpus，例如 export FLAGS_selected_gpus='0,1'。

• 设置环境变量 CUDA_VISIBLE_DEVICES，例如 export CUDA_VISIBLE_DEVICES='0,1'。

关于如何设置任务中使用的GPU设备，具体请查看 fluid. cuda_places 。

import paddle.fluid as fluid
cuda_pinned_places_cpu_num = fluid.cuda_pinned_places()
# or
cuda_pinned_places = fluid.cuda_pinned_places(1)