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.