CUDA版本的软件包未导入？

Question

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

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

我已经安装了Cuda v10，我有一个GPU。这段代码的所有需求都是通过pip一次性安装在我的虚拟环境中的

Answer 1

正如phd所指出的，它看起来像pytorch_分散检查的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