没有 CUDA 的 PyTorch torch_sparse 安装

Posted

技术标签:

【中文标题】没有 CUDA 的 PyTorch torch_sparse 安装【英文标题】:PyTorch torch_sparse installation without CUDA 【发布时间】:2021-04-27 20:11:44 【问题描述】:

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

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

用户警告:CUDA 初始化:在您的系统上找不到 NVIDIA 驱动程序。

我的系统没有 CUDA。那么没有它我怎么能安装torch_sparse 模块呢?

提前谢谢你!

亲切的问候

罗斯季斯拉夫

【问题讨论】:

【参考方案1】:

如pytorch_geometric installation instructions 中所述,您必须先安装依赖项,然后再安装torch_geometric

对于 PyTorch 1.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

【讨论】:

嗨!感谢您的答复!好吧,我已经卸载了我的```torch-geometric```,现在我正在尝试使用pip install --no-index torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.1.html 安装torch-scatter 但是我收到错误错误:找不到满足torch-scatter 要求的版本(来自版本:无)错误:即使我的 PyTorch 版本是 1.7.1 并且我没有 cuda,也没有找到火炬散射的匹配分布。你知道是什么原因吗? @Rostyslav 如果您使用 anaconda,请先使用conda install pipconda 中安装pip。之后执行这些命令。 对不起,我正在使用pip install --no-index torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.1+cpu.html,正如你所建议的那样。但仍然是同样的错误错误:找不到满足要求的版本火炬散射(来自版本:无)错误:没有找到火炬散射的匹配分布 @Rostyslav 确保您使用 Python 3.8,而不是 3.9 有效!谢谢

以上是关于没有 CUDA 的 PyTorch torch_sparse 安装的主要内容,如果未能解决你的问题,请参考以下文章

一文搞懂PyTorch与CUDA那些事

pytorch 安装中的 CUDA HOME

pytorch、显卡、显卡驱动、cuda版本是如何对应的

AIWindows配置GPU Cuda驱动和Pytorch框架

伤透了心的pytorch的cuda容器版

Pytorch RuntimeError: CUDA error: out of memory at loss.backward() , 使用 CPU 时没有错误