Run MATLAB Functions on a GPU

You can speed up your code by running MATLAB^® functions on a GPU. GPU computing in MATLAB requires Parallel Computing Toolbox™.

MATLAB Functions with `gpuArray` Arguments

Many functions in MATLAB and other toolboxes run automatically on a GPU if you supply a gpuArray data argument. A gpuArray in MATLAB represents an array that is stored on the GPU.

A = gpuArray([1 0 1; -1 -2 0; 0 1 -1]);
e = eig(A);

Whenever you call any of these functions with at least one gpuArray as a data input argument, the function executes on the GPU. The function generates a gpuArray as the result, unless returning numeric data to the local workspace is more appropriate (for example, size). You can mix inputs using both gpuArray data and arrays stored in host memory in the same function call. gpuArray-enabled functions include the discrete Fourier transform (fft), matrix multiplication (mtimes), left matrix division (mldivide), and hundreds of others.

Conditions for `gpuArray` inputs

GPU-enabled functions run on the GPU only when the input data is on the GPU. The data type of parameter arguments such as dimensions or indices do not affect where the function is run. For example, the sum function in this code runs on the GPU because the data, the first input, is on the GPU.

A = rand(10);
d = 2;
sum(gpuArray(A),d);

However, the sum function in this code does not run on GPU because the data, the first input, is not on the GPU.

A = rand(10);
d = 2;
sum(A,gpuArray(d));

Work with Complex Numbers on a GPU

If the output of a function running on a GPU could potentially be complex, you must explicitly specify its input arguments as complex. For more information, see Work with Complex Numbers on a GPU.

Work with Sparse Arrays on a GPU

The sparse function can be used to create sparse gpuArray objects. Many MATLAB functions support sparse gpuArray objects. For more information, see Work with Sparse Arrays on a GPU.

Check `gpuArray`-Supported Functions

Several MATLAB toolboxes include functions with gpuArray support. To view lists of all functions in these toolboxes that support gpuArray objects, use the links in the following table. Functions in the lists with information indicators have limitations or usage notes specific to running the function on a GPU. You can check the usage notes and limitations in the Extended Capabilities section of the function reference page. For information about updates to individual gpuArray-enabled functions, see the release notes.

Toolbox Name	List of Functions with `gpuArray` Support	GPU-Specific Documentation
MATLAB	Functions with `gpuArray` support
Statistics and Machine Learning Toolbox™	Functions with `gpuArray` support (Statistics and Machine Learning Toolbox)	Analyze and Model Data on GPU (Statistics and Machine Learning Toolbox)
Image Processing Toolbox™	Functions with `gpuArray` support (Image Processing Toolbox)	GPU Computing (Image Processing Toolbox)
Deep Learning Toolbox™	Functions with `gpuArray` support (Deep Learning Toolbox) *(see also Deep Learning with GPUs)	Scale Up Deep Learning in Parallel, on GPUs, and in the Cloud (Deep Learning Toolbox) Deep Learning with MATLAB on Multiple GPUs (Deep Learning Toolbox)
Computer Vision Toolbox™	Functions with `gpuArray` support (Computer Vision Toolbox)	GPU Code Generation and Acceleration (Computer Vision Toolbox)
Communications Toolbox™	Functions with `gpuArray` support (Communications Toolbox)	C/C++ Code Generation (Communications Toolbox)
5G Toolbox™	Functions with `gpuArray` support (5G Toolbox)
Signal Processing Toolbox™	Functions with `gpuArray` support (Signal Processing Toolbox)	Code Generation and GPU Support (Signal Processing Toolbox)
Audio Toolbox™	Functions with `gpuArray` support (Audio Toolbox)	Code Generation and GPU Support (Audio Toolbox)
Wavelet Toolbox™	Functions with `gpuArray` support (Wavelet Toolbox)	Code Generation and GPU Support (Wavelet Toolbox)
Curve Fitting Toolbox™	Functions with `gpuArray` support (Curve Fitting Toolbox)

For a list of functions with gpuArray support in all MathWorks^® products, see gpuArray-supported functions. Alternatively, you can filter by product. On the Help bar, click Functions. In the function list, browse the left pane to select a product, for example, MATLAB. At the bottom of the left pane, select GPU Arrays. If you select a product that does not have gpuArray-enabled functions, then the GPU Arrays filter is not available.

Deep Learning with GPUs

For many functions in Deep Learning Toolbox, GPU support is automatic if you have a supported GPU and Parallel Computing Toolbox. You do not need to convert your data to gpuArray. The following is a non-exhaustive list of functions that, by default, run on the GPU if available.

trainnet (Deep Learning Toolbox)
minibatchpredict (Deep Learning Toolbox)

For more information about automatic GPU support in Deep Learning Toolbox, see Scale Up Deep Learning in Parallel, on GPUs, and in the Cloud (Deep Learning Toolbox).

For custom training loops, convert your data to gpuArray and use functions with gpuArray support (Deep Learning Toolbox).

Check or Select a GPU

If you have a supported GPU, then MATLAB automatically uses it for GPU computation. If you have multiple GPUs, then you can use gpuDeviceTable to examine the properties of all GPUs detected in your system. You can use gpuDevice to select one of them, or use multiple GPUs with a parallel pool. For more information, see Identify and Select a GPU Device and Run MATLAB Functions on Multiple GPUs. To check if your GPU is supported, see GPU Computing Requirements.

gpuDeviceTable

    Index           Name           ComputeCapability    DeviceAvailable    DeviceSelected
    _____    __________________    _________________    _______________    ______________

      1      "NVIDIA RTX A5000"          "8.6"               true              true      
      2      "Quadro P620"               "6.1"               true              false

Alternatively, you can determine how many GPU devices are available, inspect some of their properties, and select a device to use from the MATLAB® desktop. On the Home tab, in the Environment area, select Parallel > Select GPU Environment.

The Parallel menu, including the Select GPU Environment pane showing two GPU devices. A tick next to the first device indicates that it is the selected device.

Use MATLAB Functions with the GPU

Open Live Script

This example shows how to use gpuArray-enabled MATLAB functions to operate with gpuArray objects. You can check the properties of your GPU using the gpuDevice function.

gpuDevice

ans = 
  CUDADevice with properties:

                 Name: 'NVIDIA RTX A5000'
                Index: 1 (of 2)
    ComputeCapability: '8.6'
          DriverModel: 'TCC'
          TotalMemory: 25544294400 (25.54 GB)
      AvailableMemory: 24734105600 (24.73 GB)
      DeviceAvailable: true
       DeviceSelected: true

  Show all properties.

Create a row vector that repeats values from -15 to 15. To transfer it to the GPU and create a gpuArray object, use the gpuArray function.

X = [-15:15 0 -15:15 0 -15:15];
gpuX = gpuArray(X);
whos gpuX

  Name      Size            Bytes  Class       Attributes

  gpuX      1x95              760  gpuArray

To operate with gpuArray objects, use any gpuArray-enabled MATLAB function. MATLAB automatically runs calculations on the GPU. For more information, see Run MATLAB Functions on a GPU. For example, use diag, expm, mod, round, abs, and fliplr together.

gpuE = expm(diag(gpuX,-1)) * expm(diag(gpuX,1));
gpuM = mod(round(abs(gpuE)),2);
gpuF = gpuM + fliplr(gpuM);

Plot the results.

imagesc(gpuF);
colormap(flip(gray));

If you need to transfer the data back from the GPU, use gather. Transferring data back to the CPU can be costly, and is generally not necessary unless you need to use your result with functions that do not support gpuArray.

result = gather(gpuF);
whos result

  Name         Size            Bytes  Class     Attributes

  result      96x96            73728  double

In general, running code on the CPU and the GPU can produce different results due to numerical precision and algorithmic differences between the GPU and CPU. Answers from the CPU and GPU are both equally valid floating point approximations to the true analytical result, having been subjected to different roundoff behavior during computation. In this example, the results are integers and round eliminates the roundoff errors.

Examples Using GPUs

Examples Running MATLAB Functions on GPUs

The following examples pass gpuArray objects to supported MATLAB functions, causing those functions to run on the GPU.

Toolbox Name	Examples
MATLAB	Sharpen an Image Using the GPU Compute the Mandelbrot Set Using GPU-Enabled Functions
Image Processing Toolbox	Perform Pixel-Based Operations on GPU (Image Processing Toolbox) Perform Thresholding and Morphological Operations on GPU (Image Processing Toolbox)
Deep Learning Toolbox	Train Network Using Custom Training Loop (Deep Learning Toolbox)
Statistics and Machine Learning Toolbox	Accelerate Linear Model Fitting on GPU (Statistics and Machine Learning Toolbox) Analyze and Model Data on GPU (Statistics and Machine Learning Toolbox)
Signal Processing Toolbox	Accelerating Correlation with GPUs (Signal Processing Toolbox) Denoise EEG Signals Using Differentiable Signal Processing Layers (Signal Processing Toolbox)
Audio Toolbox	Accelerate Audio Machine Learning Workflows Using a GPU (Audio Toolbox) Accelerate Audio Deep Learning Using GPU-Based Feature Extraction (Audio Toolbox)
Communications Toolbox	Accelerate Simulation Using GPUs (Communications Toolbox)
Wavelet Toolbox	GPU Acceleration of Scalograms for Deep Learning (Wavelet Toolbox) Wavelet Time Scattering with GPU Acceleration — Spoken Digit Recognition (Wavelet Toolbox)

Other Examples Using GPUs

The following examples make use of other automatic GPU support.

Toolbox Name	Examples
Deep Learning Toolbox	Retrain Neural Network to Classify New Images (Deep Learning Toolbox) Train Network Using Automatic Multi-GPU Support (Deep Learning Toolbox)
Communications Toolbox	Accelerate SISO Single-Carrier Link Simulation Using GPU (Communications Toolbox) Using GPUs to Accelerate Turbo Coding Bit Error Rate Simulations (Communications Toolbox)

Acknowledgments

MAGMA is a library of linear algebra routines that take advantage of GPU acceleration. Linear algebra functions implemented for gpuArray objects in Parallel Computing Toolbox leverage MAGMA to achieve high performance and accuracy.