TargetDevice

is an option for certain functions that specifies whether CPU or GPU computation should be attempted.

Details

Typically possible settings are "CPU" and "GPU".
TargetDevice can be used with NetTrain to determine on which device a network is trained.
TargetDevice can be supplied when applying a trained neural net to an input to determine on which device the network is evaluated.
With the setting TargetDevice->"GPU", the Wolfram Language will attempt to use a GPU if it is available on your computer.
TargetDevice{"GPU",n} allows a specific GPU to be used, where n is an integer between 1 and the number of GPUs available on your computer.
TargetDevice->{"GPU",All} specifies that all available GPUs should be used jointly.
TargetDevice->{"GPU",{n₁,n₂,…}} specifies that a specific subset of the GPUs should be used jointly.
Only NVIDIA GPUs with the following compute capabilities are currently supported:

In a fresh Wolfram Language installation, TargetDevice->"GPU" and related GPU settings will automatically download additional libraries ("MXNetResources"). Such a download can also trigger again in case updates are available. The download can also be started manually by running PacletInstall["MXNetResources"].

open allclose all

Train a net using the system's default GPU:

Evaluate the resulting net using the system's default GPU:

Train a net using a specified GPU:

Train a net using two specified GPUs, where each GPU receives a batch of 16 training examples per training iteration:

Train a net using all available GPUs:

If the system GPU is not supported, the operation will fail:

Top