When using the PyTorch, if you want to use the GPU only need to add ‘.cuda()’ in the corresponding place, or do you need other instructions in addition to this operation?
Why, when I only use ‘.cuda()’, I can be sure that he is running on the GPU, but I can’t feel the speed increase.