When I use a=a.cuda() or a=a.to(), the program seems to stop running,and the gpus are not used.The environment is ubuntu 16.04,pytorch 1.1 and cuda 9.0.What`s the problem?
When I use a=a.cuda() or a=a.to(), the program seems to stop running,and the gpus are not used.The environment is ubuntu 16.04,pytorch 1.1 and cuda 9.0.What`s the problem?