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

[seigot]

torch.cuda.is_available()はTrueであるものの、to be on the same device ということでおそらく計算時にcpu/cudaどちらかで計算するように寄せる必要がありそうなメッセージにみえました。
https://discuss.pytorch.org/t/expected-all-tensors-to-be-on-the-same-device-but-found-at-least-two-devices-cuda-0-and-cpu-error/151314/7
Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! error

Move alpha to the same device.

理想的にはどちらもcudaにするのが良さそうではあるのですが、回避策としてcpuに寄せるのも一つの手ではと思いました。（とても重いconvolution等でなければ、、）
具体的には以下のような感じでcudaを使わないようにコメントアウトしたらcpuで動いてくれたりしないでしょうかね。。
https://raw.githubusercontent.com/seigot/tools/master/tetris/20220315_disable_cuda.patch

•        if torch.cuda.is_available():
  

•            next_states = next_states.cuda()
  

+# if torch.cuda.is_available():
+# next_states = next_states.cuda()

[gitauto-ai]

Hey, I'm a bit lost here! Not sure which file I should be fixing. Could you give me a bit more to go on? Maybe add some details to the issue or drop a comment with some extra hints? Thanks!

Have feedback or need help?
Feel free to email [email protected].