Вы находитесь на странице: 1из 2

Printed from the Complete Wolfram Language Documentation 1

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.

See Also

Use CUDA on an External GPU on Mac

Related Functions

NetTrain ▪ TargetDevice

Related Guides

CUDALink

©1988–2019 Wolfram Research, Inc. All rights reserved. http://reference.wolfram.com/language


Printed from the Complete Wolfram Language Documentation 2

Related Tech Notes

CUDALink User Guide

©1988–2019 Wolfram Research, Inc. All rights reserved. http://reference.wolfram.com/language

Вам также может понравиться