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

GPU acceleration #31

Open
camsooper opened this issue Jun 6, 2023 · 3 comments
Open

GPU acceleration #31

camsooper opened this issue Jun 6, 2023 · 3 comments
Labels
backend Flask backend/HTTP responses cloud About cloud deployment investigation Tasks that need exploration/research
Milestone

Comments

@camsooper
Copy link

  • Which cloud platform will support this?
  • Can it be used for featurisation
  • Can it be used for encoding
  • (Or for UMAP calculation if we decide to add that?
@rmdocherty rmdocherty added cloud About cloud deployment backend Flask backend/HTTP responses investigation Tasks that need exploration/research labels Jun 7, 2023
@rmdocherty
Copy link
Contributor

  1. There are some providers (AWS, GCP, Oblivius (?)) who provision VMs with GPUS but that will incur much higher costs than a server less consumption model

  2. Depending on the features used (Blurs, Sobel definitely - others unclear) yes. If I can find a way to map a function across a footprint kernel (I.e min/max/median) then could implement most.
    I believe Ilastix uses a reduced feature set (Gaussian, Sobel and Hessian) compared to Weka and these might all be relatively straightforward (except hessian) on GPU

  3. Yes - it’s a major draw of the GPU model. The SAM embedding computation would be effectively instant on a good GPU w/ sufficient VRAM

  4. Depends. GPUMAP (fork of UMAP with GPU support) hasn’t been updated to latest CuPy, but I think it’s literally just one function/line that needs fixing. I would need to go in depth into UMAP and CuPy to fix though.

@rmdocherty
Copy link
Contributor

This looks very promising for Python GPU featurisations:
https://github.com/clEsperanto/pyclesperanto_prototype/

@rmdocherty rmdocherty added this to the SAMBA1.5 milestone Jul 20, 2023
@rmdocherty
Copy link
Contributor

Plan:

  1. Add option to GPU accelerate SAM encoding
  2. Make new branch with option to a) use pyclesperanto featurisation (need to write this) and b) GPU XGB as ensemble classifier
  3. Consider alternate web hosting that makes use of 1. and 2.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
backend Flask backend/HTTP responses cloud About cloud deployment investigation Tasks that need exploration/research
Projects
None yet
Development

No branches or pull requests

2 participants