Zixiong Huang

icon indicating a website link https://scholar.google.com.hk/citations?user=xvPL3HIAAAAJ&hl

icon company South China University of Technology icon location Guangzhou

Results 1 comments of Zixiong Huang

Unbalanced GPU memory consumption

It may be a problem of dataloader when pin_memory flag is set to true. [pytorch/pytorch#58626](https://github.com/pytorch/pytorch/issues/58626) I solved this by adding the following code here: https://github.com/NVlabs/eg3d/blob/493076071bc400be3289dd6725ef1eef0a552a9d/eg3d/training/training_loop.py#L132 ``` torch.cuda.set_device(device) ```