search

pnlbwh / pnlNipype

PNL pipeline for NIFTI images
Other
10 stars 15 forks source link

More check required to determine usability of eddy_cuda #78

Closed tashrifbillah closed 3 years ago

tashrifbillah commented 3 years ago

These checks are not sufficient to determine if eddy_cuda can be used. We should add the following to them:

    import GPUtil
    DEVICE_ID_LIST = GPUtil.getFirstAvailable()
    DEVICE_ID = DEVICE_ID_LIST[0]

The idea is inspired by this, however, imposes a new package requirement--GPUtil.

cc @sbouix