如何使用第二个GPU进行训练?
Created by: qianledan
if args.use_gpu:
# NOTE: for multi process mode: one process per GPU device.
gpu_id = 0
if os.getenv("FLAGS_selected_gpus"):
gpu_id = int(os.getenv("FLAGS_selected_gpus"))
place = core.CUDAPlace(gpu_id) if args.use_gpu else core.CPUPlace()
有两个问题:1、以上代码中FLAGS_selected_gpus参数找遍网络都没有相关的说明,哪里能提供文档资料说明?2、分布训练ps模式,单机双gpu,第一个trainer使用第一颗gpu是否使用place = core.CUDAPlace(0)?第二个trainer使用第二个GPU,是否使用place = core.CUDAPlace(1)代码?我在运行时提示出错,说place = core.CUDAPlace( )里的参数只能[0,1)之间。