Run Neural Network Training on GPUs

Specify that neural net training should use the GPU with the TargetDevice option (on Mac, recent Macs require an external GPU for this to work):

  • Currently, only NVIDIA graphics cards are supported by NetTrain.
  • To configure an external GPU for use on Mac, follow the steps in the workflow; see Use CUDA on an External GPU on Mac. However, it is not necessary to download and use CUDALink if you only wish to train a neural network with NetTrain.
  • It is necessary to have up-to-date drivers for your graphics card in order to take advantage of GPU training.
  • NetTrain will give an error if you specify TargetDevice"GPU" when no compatible GPU is available, or your graphics drivers are out of date.