Open ChaiShiMei opened 10 months ago
That's strange. First, perhaps try uninstalling pykeops to be safe (I don't use it anymore). Second, check if the extension exists with pip list | grep cauchy
. If it does, try reinstalling with pip uninstall -y cauchy-mult && pip install .
(inside the extensions folder), and then check again if pip finds it. Then try opening a Python REPL and checking if you can import the extension (import cauchy_mult
).
Your environment looks very out of date (e.g. python 3.7, pytorch 11.3)
I installed cauchy-mult following the description "going to extensions/cauchy/ and running
python setup.py install
.". But when I run the 'sashimi.py' to investigate its structure. This message shows again "CUDA extension for cauchy multiplication not found. Install by going to extensions/cauchy/ and runningpython setup.py install
. This should speed up end-to-end training by 10-50%" I installed pykeops==1.5 first.