Skip to content

Possible feature: add GPU Direct Storage Support for Nibabel #1401

Open
@yiheng-wang-nv

Description

@yiheng-wang-nv

Hi Nibabel maintainers,

This is Yiheng Wang from NVIDIA, I did some research on how to use Kvikio + Nibabel to load .nii images into GPU directly, it helps accelerate the loading time especially on large (100MB+) images.
A solution is confirmed work:

  1. using kvikio to load .nii file as bytes from disk into gpu
  2. using nibabel to load headers of the file from disk into cpu
  3. parse bytes according to headers.

Related discussion:
#1385

In the Kvikio side, I already added a related tutorial: https://github.com/rapidsai/kvikio/blob/branch-25.04/notebooks/medical_nifti_image_loading_example.ipynb

I'm wondering if it can be a feature directly in nibabel repo. What do you think?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions