search

uncbiag / mermaid

Image registration using pytorch
Other
177 stars 28 forks source link

force registration running on CPU #147

Closed colddie closed 2 years ago

colddie commented 2 years ago

Hi,

How can I force the register_image on CPU? By default, example_simple_interface.py seems always run on GPU, even if CUDA_ON was set to false.

Thank you

hbgtjxzbbx commented 2 years ago

Hi colddie thanks for the feedback. sorry for the confusion, you can set CUDA_ON to False, and comment line 41 Then code should run on CPU.