openclimatefix/nowcasting_utils

BUG: RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu

peterdudfield opened this issue · 0 comments

peterdudfield commented

Describe the bug
test_loss.py fails when running on Leonardo (a machine with GPU)
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

To Reproduce
Steps to reproduce the behavior:

  1. Log onto Leonardo (or other machine with GPU)
  2. clone repo
  3. run pytest