Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[ENH] add --prefer-gpu #725

Open
mattcieslak opened this issue Apr 19, 2024 · 2 comments
Open

[ENH] add --prefer-gpu #725

mattcieslak opened this issue Apr 19, 2024 · 2 comments
Labels
enhancement New feature or request

Comments

@mattcieslak
Copy link
Collaborator

v0.21 has gpu-enabled versions of synthstrip, synthseg, drbuddi and eddy included. Previously only eddy had gpu capabilities and these had to be accessed by a special eddy config json file.

I propose adding a --prefer-gpu flag that will switch each of these to their gpu version. There will still have to be an eddy config json file for specifying all the other options, but the "use_cuda" option will be taken out of the json file and specified on the commandline.

@mattcieslak mattcieslak added the enhancement New feature or request label Apr 19, 2024
@mattcieslak
Copy link
Collaborator Author

synthseg needs a smarter bounding box workflow like the one already being used by synthstrip. It doesn't fit on an 8gb gpu without some cropping

@mattcieslak
Copy link
Collaborator Author

Or maybe it would be better to have a --gpu [require|try|none] where require mode the will raise an error if the gpu version fails, try will try to use the gpu and fall back to the cpu version if the gpu fails and none won't attempt to use a gpu at all?

I'm just starting to run gpu jobs on an HPC and it looks like they randomly fail a lot

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant