Open 4vicii opened 1 year ago
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?
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?