主要内容

本页采用了机器翻译。点击此处可查看最新英文版本。

MATLAB 中的 GPU 计算

通过在 GPU 上运行 MATLAB® 函数来加快代码执行速度

如果 GPU 支持您想要使用的所有函数,您只需使用 gpuArray 将输入数据传输到 GPU 即可。要开始使用 GPU 计算,请参阅在 GPU 上运行 MATLAB 函数

对于深度学习,MATLAB 为多个 GPU 提供自动并行支持。请参阅Deep Learning with MATLAB on Multiple GPUs (Deep Learning Toolbox)

您可以使用 gpuDevice 函数检查并选择您的 GPU,而使用 gpuDeviceTable 函数检查多个 GPU。

函数

全部展开

gpuArray存储在 GPU 上的数组
gather将分布式数组、Composite 对象或 gpuArray 对象传输到本地工作区
gpuDevice查询或选择 GPU 设备
GPUDeviceManagerGPU 设备管理器
validateGPU验证 GPU 设备 (自 R2024b 起)
gpuDeviceCount现有 GPU 设备的数量
gpuDeviceTableGPU 设备属性 (自 R2021a 起)
wait (GPUDevice)等待 GPU 计算完成
gputimeit在 GPU 上运行函数所需的时间
existsOnGPU确定 GPU 上是否提供 gpuArray 或 CUDAKernel
isgpuarray判断输入是否为 gpuArray
arrayfun将函数应用于 GPU 上的数组的每个元素
pagefun将函数应用到分布式或 GPU 数组的每一页
gpurng控制 GPU 上的随机数生成
reset重置 GPU 设备并清除其内存
parallel.gpu.RandStreamGPU 上的随机数流
parallel.gpu.enableCUDAForwardCompatibility查询和设置 GPU 设备的前向兼容性

主题

在 GPU 上运行 MATLAB 代码

提高 GPU 性能

了解更多

相关信息

精选示例