search

rsummers11 / CADLab

Imaging Biomarkers and Computer-Aided Diagnosis Laboratory
https://www.cc.nih.gov/meet-our-doctors/rsummers.html
432 stars 186 forks source link

Error on multi-GPU when running MULAN #43

Open studabyd opened 3 years ago

studabyd commented 3 years ago

Hello, I run your project MULAN_universal_lesion_analysis on 2 Titan GPUs, but I found the error that "RuntimeError: Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight'; but device 0 does not equal 1 (while checking arguments for cudnn_convolution)" like the picture I send. I found the error is in densenet_custom_trunc.py:x = self.conv0(x). I debuged it and found that self.conv() is in device0 and copyed to device1. But x is in device0, it does't be placed in device1. So could you apply the code on multi-gpu and conld you help me with the error. Thank you! 2020-12-07 19-59-04屏幕截图

viggin commented 3 years ago

Hi studabyd, we haven't adapted and run MULAN in a multi-GPU environment yet. Maybe you need to go through the code and do some modification for that. Thanks.