search

CompImg / LST-AI

LST-AI - Deep Learning Ensemble for Accurate MS Lesion Segmentation
https://doi.org/10.1016/j.nicl.2024.103611
MIT License
18 stars 4 forks source link

How can I use LST-AI without GPU? #9

Closed sj-choi closed 6 months ago

sj-choi commented 6 months ago

I tried to run LST-AI. However, I got GPU related error message. How I can I run this with out GPU, but CPU only? Is there any options in the python script not to use GPU?

Thank you, -SC

jqmcginnis commented 6 months ago

Hi @sj-choi

There is :+1: Please specify via the --device option. Here is the description on how to use the flag:

--device: Provide an integer value (e.g. 0) for a GPU ID or cpu if you do not have access to a GPU.

Does this help?

sj-choi commented 6 months ago

Thank you very much!