TomocuPy – efficient GPU-based tomographic reconstruction with asynchronous data processing
Fast 3D data analysis and steering of a tomographic experiment by changing environmental conditions or acquisition parameters require fast, close to real-time, 3D reconstruction of large data volumes. Here a performance-optimized TomocuPy package is presented as a GPU alternative to the commonly used central processing unit (CPU) based TomoPy package for tomographic reconstruction. TomocuPy utilizes modern hardware capabilities to organize a 3D asynchronous reconstruction involving parallel read/write operations with storage drives, CPU–GPU data transfers, and GPU computations. In the asynchronous reconstruction, all the operations are timely overlapped to almost fully hide all data management time. Since most cameras work with less than 16-bit digital output, the memory usage and processing speed are furthermore optimized by using 16-bit floating-point arithmetic. As a result, 3D reconstruction with TomocuPy became 20–30 times faster than its multi-threaded CPU equivalent. Full reconstruction (including read/write operations and methods initialization) of a 2048 3 tomographic volume takes less than 7 s on a single Nvidia Tesla A100 and PCIe 4.0 NVMe SSD, and scales almost linearly increasing the data size. To simplify operation at synchrotron beamlines, TomocuPy provides an easy-to-use command-line interface. Efficacy of the package was demonstrated during a tomographic experiment on gas-hydrate formation in porous samples, where a steering option was implemented as a lens-changing mechanism for zooming to regions of interest.
- Research Organization:
- Argonne National Laboratory (ANL), Argonne, IL (United States)
- Sponsoring Organization:
- USDOE Office of Science (SC), Basic Energy Sciences (BES); USDOE Laboratory Directed Research and Development (LDRD) Program
- Grant/Contract Number:
- AC02-06CH11357; 2023-0104
- OSTI ID:
- 1899814
- Alternate ID(s):
- OSTI ID: 2229869
- Journal Information:
- Journal of Synchrotron Radiation (Online), Journal Name: Journal of Synchrotron Radiation (Online) Vol. 30 Journal Issue: 1; ISSN 1600-5775
- Publisher:
- International Union of Crystallography (IUCr)Copyright Statement
- Country of Publication:
- Denmark
- Language:
- English
Similar Records
Integration of TomoPy and the ASTRA toolbox for advanced processing and reconstruction of tomographic synchrotron data
tomoCAM : fast model-based iterative reconstruction via GPU acceleration and non-uniform fast Fourier transforms