DDP-Tutorial icon indicating copy to clipboard operation
DDP-Tutorial copied to clipboard

ENV LOCAL_RANK

Open CQ981001 opened this issue 10 months ago • 0 comments

你好,请问在env模式中,是如何获得arg.local_rank这个参数的。在启动方式中并没有看到。Tutorial中说可以通过环境变量自动获取,难道不需要添加这一条代码吗?gpu_id = int(os.environ['LOCAL_RANK'])

CQ981001 avatar Mar 28 '24 08:03 CQ981001