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 并且我有一个 GPU。这段代码的所有要求都是通过 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 的建议，从源代码安装可能会为您提供更多信息。