• C
    [cherry pick]Set FLAGS_selected_gpus for spawn (#29962) (#30097) · cda7397f
    Chen Weihang 提交于
    Set FLAGS_selected_gpus for spawn.
    
    When the child process starts, it will inherit the configuration of the main process and set the FLAGS once, but the environment variable has not been set at this time, which leads to the FLAGS_selected_gpus is keep same with mainprocess(usually empty), so manually update the flags here.
    
    注:增加了一个单测,又移除了,单测打印显示CI机器nvidia-smi只有两张卡,需要大于两张卡才能测这个问题
    cda7397f
spawn.py 16.6 KB