包的CUDA版本不导入吗？

Question

首先，我安装了torch 1.1.0，然后安装了它的依赖项。因此，我可以导入torch_scatter 1.2.0，但是在导入torch_scatter.scatter_cuda时会出现以下错误：

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

我已经安装了Cuda v10，并且我有一个图形处理器。这段代码的所有需求都是在我的虚拟环境中通过pip一次性安装在一起的。

Answer 1

正如phd所指出的-它看起来像是pytorch_scatter的setup.py文件自动检查并使用可用的cuda安装。

同样在您使用的版本中，如所见的here

...
if CUDA_HOME is not None:
    ext_modules += [
                   CUDAExtension('torch_scatter.scatter_cuda',
                  ['cuda/scatter.cpp', 'cuda/scatter_kernel.cu'])
    ]
...

这可能是一个CUDA_HOME是否可用的问题。

从源代码安装可能会为您提供建议的here的更多信息。