Add device as a parameter to indicate in which gpu to run

[xrafael]

Hi @JoHof ,

Do you think is it interesting to add a new parameter in the init function of the LMInferer class to allocate in which gpu to perform the lung segmentation? (e.g. device="cuda:1")

Thank you for this great work!
-Xavier

[JoHof]

Hi Xavier,

thanks for the suggestion. Yes, I think it could be useful in some situations. Could you create a PR for it?