如何使用 NVIDIA GPUDirect Storage 在 Python 中将数据读取至 GPU 显存

在 Python 中利用 NVIDIA GPUDirect Storage 可将数据直接读取至 GPU 显存，以解决日益严峻的存储传输瓶颈。2025 Stanford AI Index 显示，知名 AI 模型的算力约每 5 个月翻倍一次，训练 LLM 的数据集规模约每 8 个月翻倍一次。随着计算与数据规模同步高速增长，高效从存储端搬运数据成为 GPU 应用的关键环节。