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.
- 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.