Did you manage to solve your issue?
Is it related to:
"RuntimeError: CUDA driver error: initialization error"


  
    
    
    What does "RuntimeError: CUDA driver error: initialization error" mean? 
  
  
    I’m getting this error: 
"RuntimeError: CUDA driver error: initialization error"

But I can’t find what it means (or why it happens). 
Of course I want to solve it but I want to know what the error means in the first place first.