RuntimeError: Expected scalars to be on CPU, got cuda:0 instead

@Denys_Ashikhin May be close this issue since it’s fixed in Ray 2.5, and we are in the process of releasing it in the coming days?

1 Like