bioimage-io / JDLL

The Java library to run Deep Learning models
https://github.com/bioimage-io/JDLL/wiki
Apache License 2.0
27 stars 6 forks source link

Support choosing which GPU to use #35

Open ctrueden opened 11 months ago

ctrueden commented 11 months ago

As suggested by @axtimwalde at today's BDV/ImgLib2 meeting: it would be nice if JDLL had engine-agnostic API for choosing which GPU to use, when a machine has access to multiple GPU options.