Giters

xmengli / H-DenseUNet

TMI 2018. H-DenseUNet: Hybrid Densely Connected UNet for Liver and Tumor Segmentation from CT Volumes

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

multi GPU problem

JohnleeHIT opened this issue · comments

commented

Hello, I meet some problem when training your code, I debuged the code and find the problem, it attribute to the function "make_parallel", once i skip the process and use just one gpu, no error. However, when add this item, I get :
"could not set cudnn tensor descriptor: CUDNN_STATUS_BAD_PARAM"
Any idea why this happens? Thanks.