Friday, 7 May 2021

CUDA version of package not importing?

Firstly, i installed torch 1.1.0, and then I installed its' dependencies. So, I can import torch_scatter 1.2.0 however i get this error when importing torch_scatter.scatter_cuda:

 import torch_scatter.scatter_cuda
ModuleNotFoundError: No module named 'torch_scatter.scatter_cuda'

I have cuda v10 installed and I have a GPU. All of the requirements for this code were installed together through pip in one go on my virtual environment.



from CUDA version of package not importing?

No comments:

Post a Comment