search

deephealthproject / eddl

European Distributed Deep Learning (EDDL) library. A general-purpose library initially developed to cover deep learning needs in healthcare use cases within the DeepHealth project.
https://deephealthproject.github.io/eddl/
MIT License
34 stars 10 forks source link

Problems with multi-GPU #222

Closed salvacarrion closed 3 years ago

salvacarrion commented 3 years ago

Describe the bug Only one GPU is used when a model is run in multi-gpu mode

To Reproduce

  1. Use CS_GPU({1,1})
  2. Check with nvidia-smi

Expected behavior Use of selected GPUs

Desktop (please complete the following information):

Hint Could be related to this line: https://github.com/deephealthproject/eddl/blob/f754672fe491322acb8c4b18393743baa8129459/src/net/net_api.cpp#L159

RParedesPalacios commented 3 years ago

I think this is solved now in Master in the new release. We remove the mop taskloop and use the standard amp parallel for