reconstrue / brightfield

Reconstruction of biocytin-stained neurons detected in brightfield microscopy image stacks
http://reconstrue.com
Apache License 2.0
2 stars 1 forks source link

Docs: Leveraging the TPU/GPUs available on Colab #16

Open JohnTigue opened 4 years ago

JohnTigue commented 4 years ago

On Colab, you have to explicitly request a GPU. As of late 2019, there are three that have been seen recently

The first two work with Rapids, the K80 does not. Let's make sure we have one that works with Rapids.

In the first pass of getting ShuTu working on Colab, there is no focus on the Nvidia T4 GPU that is made freely available on Colab. ShuTu is not written to leverage GPUs, rather it uses multiple CPU cores, of which there is only one on Colab (cue sad trombone).

Nonetheless, the whole imaging pipeline should run on RAPIDs and CUDA etc. Just chuck the image stack onto the GPU and out comes a SWC. Yup, just that simple :)

JohnTigue commented 4 years ago

Nvidia: GPU Accelerated Computing with Python

JohnTigue commented 4 years ago

See #18 for notes on GPUing max intensity projections, which would be trivial.

JohnTigue commented 4 years ago

Speeding up Python is a Colab hosted Jupyter notebook.