AllenNeuralDynamics / AllenNeuralDynamics/aind-large-scale-cellpose

Improve performance in GPU when processing data

未关闭
#2 1 条评论 1 个 reaction 已指派 1 人 已被 @camilolaiton 认领 在 GitHub 查看
enhancement
主要语言
Python
星标
10
派生
0
PR 合并指标
30 天内没有已合并 PR

描述

The current approach assumes that there's a single GPU processing data. However, this is not true for other users or even us when we need to scale to bigger datasets. Therefore, I need to implement a multiprocessing approach that uses all the GPUs available at the moment.

In addition to that, there's a down time in the GPU processing that is wasted. With a Z1 dataset in the 2nd multiscale, this down time is about 12 seconds when we're writing to the zarr dataset in the prediction of the gradients. The GPU should, ideally, be processing data all the time while another process should be looking at gathering the results and writing them to zarr.

The following image is an example of this:
![image](https://github.com/AllenNeuralDynamics/aind-z1-cell-segmentation/assets/36769694/845cf738-b70e-477e-b81c-39ff2cbfeb16)

贡献指南

这个仓库没有索引到贡献指南

评估

这个 Issue 还没有评估数据。

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。