python check cuda device

So it looks like the CUDA device is not being recognized. Could you please try this

from tensorflow.python.client import device_lib

device_lib.list_local_devices()


https://github.com/ludwig-ai/ludwig/issues/365