i make the model and the input tensor in GPU by model = model.cuda() input_tensor = input_tensor.cuda()
But the Volatile GPU-util is 0% with i occupy the video memory?
If i use cpu to train ,why does it occupy the Video Memory?
And what should i do to use Gpu to train?
Thank you very much!
