DeepLabV3Plus-Pytorch icon indicating copy to clipboard operation
DeepLabV3Plus-Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon VainF

distributed training error

Open 4vicii opened this issue 1 year ago • 0 comments

Hi，I have 8 GPU(GTX 4090).when i set the --gpu_id as '0,1,2,3,4,5,6,7' , the process will be stucked in outputs = model(images) Unless ,I set one GPU how can i fix the problem?

my env
torch 2.0 cuda 12.0

4vicii avatar Sep 18 '23 08:09 4vicii