device_cn.rst 823 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37

设备管理
==================


paddle_num_threads
*******************************************
(始于0.15.0)

控制每个paddle实例的线程数。

取值范围
---------------
Int32型,缺省值为1。

示例
-------
FLAGS_paddle_num_threads=2 - 将每个实例的最大线程数设为2。


selected_gpus
*******************************************
(始于1.3)

设置用于训练或预测的GPU设备。

取值范围
---------------
以逗号分隔的设备ID列表,其中每个设备ID是一个非负整数,且应小于您的机器拥有的GPU设备总数。

示例
-------
FLAGS_selected_gpus=0,1,2,3,4,5,6,7 - 令0-7号GPU设备用于训练和预测。

注意
-------
使用该flag的原因是我们希望在GPU设备之间使用聚合通信,但通过CUDA_VISIBLE_DEVICES只能使用共享内存。