DIGITS Error: Check failed: error == cudaSuccess (30 vs. 0) unknown error

I have follow the steps to install Nvidia CUDA 10.2 and cuDNN 7 for 10.2 from:


https://developer.nvidia.com/rdp/cudnn-download
Then I have follow the steps in the two day demo (for the Jetson) to try to set DIGITS on my computer from:

I am running Ubuntu 18.04

Everything was going well until I try to train a model.
I get the following error:
Check failed: error == cudaSuccess (30 vs. 0) unknown error

I have checked online but I have not found any solution.
Running nvidia-smi gives me the following:

Another problem I am seen which may or may not be related is that when I run deviceQuery I get the following:
:/usr/local/cuda/samples/bin/x86_64/linux/release$ ./deviceQuery
./deviceQuery Starting…

CUDA Device Query (Runtime API) version (CUDART static linking)

cudaGetDeviceCount returned 999
-> unknown error
Result = FAIL

I have seen someplace that mention to run deviceQuery as sudo helps but it is not my case.

Any information would be much appreciated.