Traceback (most recent call last):
File “/home/cs20d010/Elisha/PREF-SHAP/debug_train_model.py”, line 107, in
c.train_model()
File “/home/cs20d010/Elisha/PREF-SHAP/utils/utils.py”, line 375, in train_model
results = self.SGD_krr_loop()
File “/home/cs20d010/Elisha/PREF-SHAP/utils/utils.py”, line 352, in SGD_krr_loop
best_model,best_tr,best_val,best_test = self.model(train_X, train_y, val_X, val_y, test_X, test_y, inp, 1e-5,self.m)
File “/home/cs20d010/Elisha/PREF-SHAP/GPGP/KRR_model.py”, line 127, in SGD_KRR
return train_krr(kernel,Xtrain,Ytrain,Xval,Yval,Xtest,Ytest,pen,m_fac)
File “/home/cs20d010/Elisha/PREF-SHAP/GPGP/KRR_model.py”, line 97, in train_krr
loss = model(Xtrain, Ytrain)
File “/home/cs20d010/.local/lib/python3.10/site-packages/torch/nn/modules/module.py”, line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File “/home/cs20d010/.local/lib/python3.10/site-packages/torch/nn/modules/module.py”, line 1562, in _call_impl
return forward_call(*args, **kwargs)
File “/home/cs20d010/.local/lib/python3.10/site-packages/falkon/hopt/objectives/exact_objectives/new_compreg.py”, line 32, in forward
L, A, AAT, LB, c = self._calc_intermediate(X, Y)
File “/home/cs20d010/.local/lib/python3.10/site-packages/falkon/hopt/objectives/exact_objectives/new_compreg.py”, line 56, in _calc_intermediate
kmn = self.kernel(self.centers, X)
File “/home/cs20d010/.local/lib/python3.10/site-packages/falkon/kernels/kernel.py”, line 183, in call
return mm_impl(
File “/home/cs20d010/.local/lib/python3.10/site-packages/falkon/mmv_ops/fmm.py”, line 712, in fmm
return KernelMmFnFull.apply(kernel, opt, kwargs_m1, kwargs_m2, out, diag, X1, X2, *kernel.diff_params.values())
File “/home/cs20d010/.local/lib/python3.10/site-packages/torch/autograd/function.py”, line 574, in apply
return super().apply(*args, **kwargs) # type: ignore[misc]
File “/home/cs20d010/.local/lib/python3.10/site-packages/falkon/mmv_ops/fmm.py”, line 630, in forward
out = KernelMmFnFull.run_gpu_gpu(X1, X2, out, kernel, comp_dtype, opt, False, kwargs_m1, kwargs_m2)
File “/home/cs20d010/.local/lib/python3.10/site-packages/falkon/mmv_ops/fmm.py”, line 534, in run_gpu_gpu
gpu_info = _get_gpu_info(options, slack=options.memory_slack)
File “/home/cs20d010/.local/lib/python3.10/site-packages/falkon/mmv_ops/utils.py”, line 50, in _get_gpu_info
gpu_info = [v for k, v in devices.get_device_info(opt).items() if v.isGPU]
File “/home/cs20d010/.local/lib/python3.10/site-packages/falkon/utils/devices.py”, line 192, in get_device_info
__COMP_DATA = _get_gpu_device_info(opt, g, __COMP_DATA)
File “/home/cs20d010/.local/lib/python3.10/site-packages/falkon/utils/devices.py”, line 84, in _get_gpu_device_info
mem_free, mem_total = mem_get_info(g)
File “/home/cs20d010/.local/lib/python3.10/site-packages/falkon/c_ext/init.py”, line 15, in call_cuda
from ._backend import _assert_has_ext
File “/home/cs20d010/.local/lib/python3.10/site-packages/falkon/c_ext/_backend.py”, line 86, in
torch.ops.load_library(lib_path)
File “/home/cs20d010/.local/lib/python3.10/site-packages/torch/_ops.py”, line 1295, in load_library
ctypes.CDLL(path)
File “/usr/lib/python3.10/ctypes/init.py”, line 374, in init
self._handle = _dlopen(self._name, mode)
OSError: /home/cs20d010/.local/lib/python3.10/site-packages/falkon/c_ext/_C.so: undefined symbol: _ZN2at4_ops5zeros4callEN3c108ArrayRefINS2_6SymIntEEENS2_8optionalINS2_10ScalarTypeEEENS6_INS2_6LayoutEEENS6_INS2_6DeviceEEENS6_IbEE
I don’t know what falkon is but these types or errors are usually raised if your libraries are not compatible with each other, i.e. in this case probably falkon and torch.