search

cwmok / Conditional_LapIRN

Conditional Deformable Image Registration with Convolutional Neural Network
MIT License
79 stars 16 forks source link

Python and PyTorch versions #12

Closed annareithmeir closed 1 year ago

annareithmeir commented 1 year ago

Have you tested the code with Python=3.10 and torch=2.0.1 by any chance? If not, what are the latest versions the code has been tested - still Python=3.5.2 and torch=1.7.1?

cwmok commented 1 year ago

I have tested this codebase with the latest version python and pytorch. If you found that the loss cannot decrease, you may switch the pytorch with different cuda version. In my case, pytorch with cu118 is problematic, but pytorch with cu117 works flawlessly.

annareithmeir commented 1 year ago

Thanks, I got it to work with python 3.10, torch 2.0.1 and cu117