不使用 CUDA 的 PyTorch torch_sparse 安装

Question

我是 PyTorch 的新手，我遇到了一个问题，即我无法正确安装我的 torch_sparse 模块。\n一般来说，我想使用模块torch_geometric- 我已经安装了这个模块。但是，在程序执行期间，我不断收到错误 ModuleNotFoundError: No module named \xe2\x80\x98torch_sparse\xe2\x80\x99 。

\n

我尝试安装它，但是当我pip install torch-sparse在 anaconda 中使用该命令时，出现错误：

\n

用户警告：CUDA 初始化：发现您的系统上没有 NVIDIA 驱动程序。

\n

我的系统没有 CUDA。那么如果没有它我该如何安装torch_sparse模块呢？

\n

先感谢您！

\n

亲切的问候

\n

罗斯季斯拉夫

\n

Answer 1

正如pytorch_geometric 安装说明中所述，您必须先安装依赖项，然后torch_geometric再安装依赖项。

对于 PyTorch1.7.0和 CPU：

pip install --no-index torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cpu.html
pip install --no-index torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.0+cpu.html
pip install --no-index torch-cluster -f https://pytorch-geometric.com/whl/torch-1.7.0+cpu.html
pip install --no-index torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.7.0+cpu.html
pip install torch-geometric

请注意torch-1.7.0+cpu每页的最后