Hi! When I try to run your code on my second GPU, I meet this problem:
RuntimeError: Expected all tensors to be on the same device, but found at least two devices: cuda:1 and cuda:0!
I've mentioned it in issue #50 and I managed to fix it by the revision below:
And the task will run on the assigned GPU (can be viewed by command nvidia-smi).
Please let me know if the revision looks good for you.
Hi! When I try to run your code on my second GPU, I meet this problem:
And the task will run on the assigned GPU (can be viewed by command
RuntimeError: Expected all tensors to be on the same device, but found at least two devices: cuda:1 and cuda:0!I've mentioned it in issue #50 and I managed to fix it by the revision below:nvidia-smi). Please let me know if the revision looks good for you.