Hello, I have a question about using pytorch with gpu. I have got torch.cuda.is_available()=true and torch.cuda.get_device_name(0) = gtx 1060, also set the cnn = cnn.cuda() and x = Variable(x)
return x.cuda() if torch.cuda.is_available() else x, but when I check the GPU usage, it is 0% during training, and the progress is so slow. When I use tensorflow or keras, I have got 20~30% gpu usage and the progress is much more faster. Am I doing anything wrong with the settings? Thanks.