I recently upgraded my main GPU to an RTX 2080 GPU and whenever I try to use it to train a Neural Network with it, Mathematica will simply exit the kernel Without error message, aborting the computation.
For example:
resource = ResourceObject["MNIST"];
trainingData = ResourceData[resource, "TrainingData"];
testData = ResourceData[resource, "TestData"];
network = NetChain[{FlattenLayer[], LinearLayer[], SoftmaxLayer[-1]},
"Output" -> NetDecoder[{"Class", Range[0, 9]}],
"Input" -> NetEncoder[{"Image", {28, 28}, "Grayscale"}]
]
trained = NetTrain[network, trainingData, ValidationSet -> testData,
MaxTrainingRounds -> 50, TargetDevice -> {"GPU", 1}];
Simply crashes the Kernel, while training the network with my secondary GPU, a GTX 750Ti, by setting {"GPU", 2} works just fine.
On the other hand, CUDA seems to be working well, CUDAQ[] returns true, $CUDADeviceCount lists my two GPUs and
CUDAImageConvolve[IMAGE, , {{-1, -2, 3}}]
will work just like in the sample page when the option
$CUDADevice = 1
is set.
Any idea on why this might be happening?
Edit: I'm running Mathematica 11.3, that I just reinstalled in case was the issue, running on Windows 10 Pro 64-bits.
Thanks.