Solved by albanD in post #2 
              
              
                Hi, 
Can you try checking the ram usage if you just do a simple cuda op like torch.rand(10, device="cuda")? The cpu size memory usage of the cuda driver is known to be very large.