AttributeError: '_OpNamespace' 'falkon' object has no attribute 'mem_get_info'

Hi,
I am trying to run Falkon with GPU and getting the below error:

File “/usr/local/lib/python3.10/dist-packages/falkon/utils/devices.py”, line 92, in _get_gpu_device_info
mem_free, mem_total = mem_get_info(g)
File “/usr/local/lib/python3.10/dist-packages/falkon/c_ext/init .py”, line 17, in call_cuda
return getattr(torch.ops.falkon, name)(*args, **kwargs)
File “/usr/local/lib/python3.10/dist-packages/torch/_ops.py”, line 569, in getattr
raise AttributeError(
AttributeError: ‘_OpNamespace’ ‘falkon’ object has no attribute ‘mem_get_info’

Please help me resolve this.

Thanks!

This error seems to be specific to the falkon repository, so I would recommend checking with the authors if this error was already seen.