search

MASILab / Synb0-DISCO

Distortion correction of diffusion weighted MRI without reverse phase-encoding scans or field-maps
https://my.vanderbilt.edu/masi
57 stars 28 forks source link

Does pipeline.sh use GPU? #53

Open tashrifbillah opened 11 months ago

tashrifbillah commented 11 months ago 
epi_reg distorted b0 to T1
epi_reg --epi=/INPUTS/b0.nii.gz --t1=/INPUTS/T1.nii.gz --t1brain=/tmp/tmp.z3zFpgMgdl/T1_mask.nii.gz --out=/tmp/tmp.z3zFpgMgdl/epi_reg_d
Running FAST segmentation
FLIRT pre-alignment
Running BBR

Shouldn't this step use GPU? nvidia-smi command does not print an associated GPU process at the time of running.

toomanycats commented 6 months ago

I'm not a developer or expert with this program. However, I'd say no. Those are just FSL programs. They use CPUs and can use multiple threads.

AFAIK, nothing in the pipeline uses GPU. The model is already built and so inference is just a forward run through the model and uses CPU. IN fact, I know that the torch library in the container version, was not compiled for GPUs.