CBICA / CaPTk

Cancer Imaging Phenomics Toolkit (CaPTk) is a software platform to perform image analysis and predictive modeling tasks. Documentation: https://cbica.github.io/CaPTk
https://www.cbica.upenn.edu/captk
Other
173 stars 63 forks source link

GPU acceleration option while deep learning inference? #1455

Closed pjw1 closed 2 years ago

pjw1 commented 2 years ago

Hi! Thank you for opening your code.

I wonder whether there exists a gpu acceleration option when running deep learning model (e.g. skull stripping in DeepMedic). I could not find any option related to gpu in DeepMedic, maunal or github issue.

Is there gpu acceleration option while deep learning inference?

Thank you in advance!