search

KCL-BMEIS / niftyreg

This project contains command line tools to perform rigid, affine and non-linear registration of nifti or analyse images as well as utilities
BSD 3-Clause "New" or "Revised" License
141 stars 42 forks source link

How to run NiftyReg on GPU #115

Open koegl opened 2 months ago

koegl commented 2 months ago

How can I run NiftyReg on my GPU?

I built it from the branch rearchitect-and-reinstate-gpu (I set cmake -USE_CUDA=ON).

When I run reg_f3d on an example .nii pair, the version from the main branch takes 12s and the one from the rearchitect-and-reinstate-gpu takes 15s - so I think it is not running on the GPU.

Is there something else that I have to do to enable the GPU?