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.