Hi, all
Why I should use same gpu number when I trained the model?
If not, this error occurs.
RuntimeError: Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight'; but device 0 does not equal 2 (while checking arguments for cudnn_convolution)