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
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