Add ability to set GPU ID

rosalindfranklininstitute / parakeet

Phantom generator

GNU General Public License v3.0

18 stars 5 forks source link

Add ability to set GPU ID #56

Closed jmp1985 closed 10 months ago

jmp1985 commented 11 months ago

Also refactor a bit of the multi processing config. In order to use multiple GPUs I will need to refactor a bit more to avoid pickling of hdf5 file handles.