Hi guys. I am a newbie in pytorch.
My Ubuntu 16.04.02 LTS got python2.7 and cuda 9.0 installed (torch.cuda.is_available() returns True). However, when I tried run the following, my python just got frozen and I can only type ctrl+z to force exit.
import torch
torch.randn(10).cuda()
Then freeze until ctrl+z.
Did I do something wrong? Anyone could help? Thanks a lot.