没有 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 pip
在conda
中安装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 安装的主要内容,如果未能解决你的问题,请参考以下文章
AIWindows配置GPU Cuda驱动和Pytorch框架
Pytorch RuntimeError: CUDA error: out of memory at loss.backward() , 使用 CPU 时没有错误