I tried to train a neural network using Pytorch. However, the code was stuck at  model.to(cuda). I am using CUDA 8.0 and try pytorch 1.0.0 and 0.4.1 with same problem. I checked the driver and it is good. And I tried the model.to(torch.device(cuda)). Could u pls help with me? Thank you very much.