Is there a way to run this on CPU instead of CUDA GPU?

cwmok / LapIRN

Large Deformation Diffeomorphic Image Registration with Laplacian Pyramid Networks

MIT License

121 stars 21 forks source link

Is there a way to run this on CPU instead of CUDA GPU? #1

Closed jcmvega5 closed 3 years ago

jcmvega5 commented 3 years ago

The machines I have access to are Windows/MacOS but neither of them have nVidia CUDA. Are there workarounds for this?

cwmok commented 3 years ago

Currently, there is no CPU implementation of LapIRN as training LapIRN could be extremely time-consuming with CPU. Yet, the testing scripts of LapIRN can be easily modified to CPU version by removing ".cuda()" and setting "use_cuda = False". I have attached these modified scripts as an example with this post. The scripts are tested under Windows 10 + Anaconda + Pytorch 1.7.1 (CPU). CPU_code.zip