ALexH0U

no more....

Results 2 comments of


                                            ALexH0U

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument mat2 in method wrapper_CUDA_mm)

me too~The problem persists

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument mat2 in method wrapper_CUDA_mm)

![微信截图_20240928183216](https://github.com/user-attachments/assets/d095aba4-f224-49ef-8c92-139e136c9c12)