当前位置：首页 > news >正文

TIGRE多GPU并行计算：如何实现任意大尺寸图像重建

news 2026/7/14 14:05:50

TIGRE多GPU并行计算：如何实现任意大尺寸图像重建

【免费下载链接】TIGRETIGRE: Tomographic Iterative GPU-based Reconstruction Toolbox项目地址: https://gitcode.com/gh_mirrors/ti/TIGRE

TIGRE（Tomographic Iterative GPU-based Reconstruction Toolbox）是一款开源的GPU加速断层图像重建工具箱，它通过多GPU并行计算技术，实现了任意大尺寸图像的快速重建。本文将详细介绍TIGRE如何利用多GPU架构突破单GPU内存限制，实现超大规模CT图像的高效重建。

为什么需要多GPU并行计算？ 🤔

在医学影像和工业CT领域，高分辨率三维图像重建面临着巨大的计算挑战。传统单GPU方案受限于显存容量，无法处理超大规模图像数据。例如，一个3340×3340×900体素的三维图像需要超过40GB的显存，这超出了大多数单张GPU的能力范围。

TIGRE通过创新的多GPU并行架构，将大尺寸图像智能分割到多个GPU上同时处理，不仅突破了显存限制，还大幅提升了计算速度。这种设计使得研究人员能够处理以前无法想象的超大规模CT数据。

TIGRE使用CGLS算法进行大规模图像重建的效果展示

TIGRE多GPU架构设计原理 🏗️

TIGRE的多GPU并行架构基于以下核心技术：

1. 智能数据分割策略

TIGRE采用自动内存管理机制，根据可用GPU显存动态分割图像数据。在Common/CUDA/ray_interpolated_projection.cu中，系统会自动计算每个GPU可以处理的图像块大小：

// 计算内存需求并决定分割策略 size_t mem_image = (size_t)geo.nVoxelX * geo.nVoxelY * geo.nVoxelZ * sizeof(float); size_t mem_proj = (size_t)geo.nDetecU * geo.nDetecV * sizeof(float); splits = mem_image / mem_free + 1; // 向上取整分割

2. 并行投影与反投影算法

TIGRE实现了多种并行投影算法，包括：

Siddon并行投影算法：Common/CUDA/Siddon_projection_parallel.cu
插值并行投影算法：Common/CUDA/ray_interpolated_projection_parallel.cu
并行体素反投影算法：Common/CUDA/voxel_backprojection_parallel.cu

3. GPU设备管理

TIGRE通过Common/CUDA/GpuIds.hpp和Common/CUDA/GpuIds.cpp中的GpuIds类来管理多GPU设备：

struct GpuIds { int* m_piDeviceIds; int m_iCount; void SetIds(int iCount, int* piDeviceIds); void SetAllGpus(int iTotalDeviceCount); bool AreEqualDevices() const; };

使用TIGRE重建的咖啡豆CT图像，展示了多GPU并行计算的高质量结果

快速上手：多GPU配置指南 🚀

步骤1：检测可用GPU

在MATLAB中，可以使用以下命令查看系统可用的GPU：

% 列出所有GPU名称 listGpuNames = getGpuNames(); disp('可用的GPU设备：'); disp(listGpuNames);

步骤2：配置多GPU环境

根据需求选择GPU设备：

% 使用所有相同型号的GPU gpuids = GpuIds(); % 仅使用特定型号的GPU（如RTX 2080 Ti） gpuids2080 = GpuIds('GeForce RTX 2080 Ti'); % 手动指定GPU设备ID gpuids_custom = GpuIds(); gpuids_custom.devices = int32([0, 2, 3]); % 使用ID为0,2,3的GPU

步骤3：运行多GPU重建

在重建算法中指定GPU配置：

% 使用多GPU进行FDK重建 imgFDK = FDK(projections, geo, angles, 'gpuids', gpuids); % 使用多GPU进行OS-SART迭代重建 imgOSSART = OS_SART(projections, geo, angles, niter, 'gpuids', gpuids);