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.
/SystemFiles/Components/MXNetLink/LibraryResources/
we see that cuda library is present in version 11.0 but not in version 11.1. $\endgroup$