Solved by albanD in post #3 
              
              
                Hi, 
You can try to actually use a cuda element to see such errors: torch.rand(1, device="cuda"). 
Does that give you the informations you want?