Error while teaching neural network Alexnet

Hello. I use Alexnet neural network for lane detection. While reteaching it with new frames of road, i have an error:

An unexpected error occurred during CUDA execution. The CUDA error was: CUDA_ERROR_LAUNCH_FAILED

I have GeForce 840M GPU. Matlab uses it for teaching.

But when i use CPU for teaching(different PC (without discrete GPU) is used), it works properly, but it takes very long time.

If have any ideas how to solve it, please tell them.

[Alison Eele]

It is possible you are encountering a kernel execution timeout which is imposed by the operating system on processes running on the GPU for longer than a few seconds. We have more information about it here: https://www.mathworks.com/matlabcentral/answers/99300

After you've encountered that error can you do any other code that interacts with the GPU successfully without restarting MATLAB? e.g.

A=gpuArray.rand(1)

If you can't then I would strongly suspect the execution timeout to be to be the problem.