WORKFLOW

Run Neural Network Training on GPUs

Specify that neural net training should use the GPU with the TargetDevice option:

  • GPU training is only available on Windows and Linux platforms with NVIDIA graphics cards. On these platforms, TargetDevice"GPU" is effectively equivalent to TargetDevice"CUDA".
  • 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.