CompactBilinearPooling-Pytorch icon indicating copy to clipboard operation
CompactBilinearPooling-Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon DeepInsight-PCALab

multi GPUs error: RuntimeError: arguments are located on different GPUs

Open JingyunLiang opened this issue 6 years ago • 3 comments

For multi GPU, it outputs:

RuntimeError: arguments are located on different GPUs at /pytorch/torch/lib/THC/generic/THCTensorMathBlas.cu:236

How to fix it?

JingyunLiang avatar May 21 '18 05:05 JingyunLiang