Solved by ptrblck in post #2 
              
              
                Is your model running fine on the CPU? 
This should give you a better stack trace than the current one. 
If it’s running fine of the CPU, could you check, if you might be running out of memory and reduce the batch size if possible, since sometimes library errors might mask an actual OOM issue?