search

JoHof / lungmask

Automated lung segmentation in CT
Apache License 2.0
669 stars 150 forks source link

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

Open xrafael opened 9 months ago

xrafael commented 9 months ago

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 commented 9 months ago

Hi Xavier,

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