It seems that
TargetDevice->"GPU" doesn't work for me in Mathematica version 11.1. I get an error message:
trained = NetTrain[lenet, trainingData, ValidationSet -> testData, MaxTrainingRounds -> 3, TargetDevice -> "GPU"] (* $Failed *)
NetTrain::trgdevos: TargetDevice -> "GPU" is not supported on MacOSX.
However, the same works for me in version 11.0. So why does it break in version 11.1? Is there a way to fix that?
I'm using a NVIDIA GeForce GT 750M on mac os version 10.12.3. My graphics card is cuDNN compatible (=3.0) according to the information here.