Pretrained DeepLabv3 and DeepLabv3+ for Pascal VOC & Cityscapes
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool
4vicii opened this issue 9 months ago · comments
I have 8 GPUs, --gpu_id '0,1,2,3,4,5,6,7',but every time stucked in the model forward outputs = model(images) and i try to use a single GPU,it works. my env torch 2.0 cuda 12.0 How can i fix this problem?
