I'm trying to run a package that uses pykeops, which I installed, on a GPU but am getting the following error. How do I resolve?
File "/home/joseph/.local/lib/python3.8/site-packages/pykeops/torch/generic/generic_red.py", line 627, in __call__
out = GenredAutograd.apply(
File "/home/joseph/.local/lib/python3.8/site-packages/pykeops/torch/generic/generic_red.py", line 78, in forward
myconv = keops_binder["nvrtc" if tagCPUGPU else "cpp"](
File "/home/joseph/.local/lib/python3.8/site-packages/keopscore/utils/Cache.py", line 68, in __call__
obj = self.cls(*args)
File "/home/joseph/.local/lib/python3.8/site-packages/pykeops/common/keops_io/LoadKeOps_nvrtc.py", line 15, in __init__
super().__init__(*args, fast_init=fast_init)
File "/home/joseph/.local/lib/python3.8/site-packages/pykeops/common/keops_io/LoadKeOps.py", line 31, in __init__
self.init_phase2()
File "/home/joseph/.local/lib/python3.8/site-packages/pykeops/common/keops_io/LoadKeOps_nvrtc.py", line 20, in init_phase2
pykeops_nvrtc = importlib.import_module("pykeops_nvrtc")
File "/usr/lib/python3.8/importlib/__init__.py", line 127, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File "<frozen importlib._bootstrap>", line 1014, in _gcd_import
File "<frozen importlib._bootstrap>", line 991, in _find_and_load
File "<frozen importlib._bootstrap>", line 973, in _find_and_load_unlocked
ModuleNotFoundError: No module named 'pykeops_nvrtc'
I'm trying to run a package that uses pykeops, which I installed, on a GPU but am getting the following error. How do I resolve?