force registration running on CPU

uncbiag / mermaid

Image registration using pytorch

Other

178 stars 29 forks source link

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.