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.