search

visinf / n3net

Neural Nearest Neighbors Networks (NIPS*2018)
Other
286 stars 45 forks source link

"import matmul_cuda" ImportError #21

Open LvZhenFeng opened 2 years ago

LvZhenFeng commented 2 years ago 
>>> import matmul_cuda
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ImportError: libcudart.so.9.0: cannot open shared object file: No such file or directory
>>>

Environment I have tried: python=3.6 pytorch=1.10.2 cudatoolkit=9.2 or 10.0 or 11.3

I followed the readme to complete the environment configuration and encountered no errors; Then I run the test statement and ImportError: libcudart.so.9.0: cannot open shared object file: No such file or directory

LvZhenFeng commented 2 years ago

I think could be the cupy version problem, it should be same as cuda (cudatoolkit)

when I use pip install cupy-cuda113, ImportError is solved.