Python CUDA版本的软件包未导入？

首先，我安装了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一次性安装的。

正如phd指出的那样-它看起来像pytorch_分散检查的setup.py文件，并自动使用可用的cuda安装

同样在您使用的版本中，如图所示：

可能是关于

CUDA_HOME

是否可用的问题

---

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

这可能是

pip

与

python

可执行文件不对应的情况，即使它位于虚拟环境中。

pip list

和

python-m pip list

是否给出了相同的输出，并且相关的包在这两个列表中是否可见？@9769953是的，它们在我看来是一样的！两者都包括所有合适的软件包。是否可以检查CUDA_HOME是否在任何地方可用？您知道如何从源代码安装吗？CUDA_HOME作为

从torch.utils.cpp_extension导入CPPPextension、CUDAExtension、CUDA_HOME

，因此您可以查看这是否适用于您。要从源代码安装，请将git存储库下载到一个文件夹中，将cd放入其中，然后使用

python setup.py install

。torch_District的作者表示，这可能会导致从PyPI安装时未遇到错误。此外，请确保您已检查CUDA是否位于您的路径中，如下所述：

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