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 mat1 in method wrapper...

[karanjakhar]

Describe the bug
I followed the steps from "how to contribute" and setting up Lugwig locally for development. When I run pytest I get

Environment (please complete the following information):

• OS: Ubuntu
• Version: 20.04
• Python version: 3.10
• Ludwig version: 0.8

[tgaddair]

Hey @karanjakhar, some of the tests have known issues when running on GPU. We have a PR in progress to fix these tests:

#3578

As a temporary workaround, can you try running these tests on CPU? It should work to do something like:

CUDA_VISIBLE_DEVICES="" pytest ...

cc @jeffkinnison

[karanjakhar]

Most of these are solved with your suggestion. But I am getting this now: