SparseTensor need torch-sparse

img


之前拷项目遇到了题目中的问题,始终找不到解决方案,后来多方调试、猜测才发现,图中这几个库缺一不可。
之前我没有安装torch-scatter,因为不知道要这样做,根据报错信息以及网上查询也没有找到解决方案,抱着试一试的心理安装了这个库,奇迹般的成功了,因此记录一下。
安装这些库网上有很多帖子,根据torch和cuda版本安装对应的whl文件,再cd到路径下pip install即可。

【以下回答由 GPT 生成】

对于SparseTensor需要安装torch-sparse库的问题,你可以按照以下步骤解决:

  1. 首先确定你已经安装了PyTorch和torch-scipy库。可以通过在终端中运行以下命令来检查它们是否已经安装:
import torch
import torch_scipy

如果没有出现错误信息,则说明这两个库都已经正确安装。

  1. 确定你的CUDA版本。可以通过在终端中运行以下命令来查看你的CUDA版本:
nvcc --version
  1. 找到适合你CUDA版本的torch-sparse库的whl文件。你可以在以下网址找到对应的whl文件: https://pytorch-geometric.com/whl/torch-1.8.0+cu102.html (假设你的CUDA版本是102)。

  2. 下载符合你CUDA版本的torch-sparse库的whl文件,并通过cd命令进入该文件所在路径。

  3. 在终端中输入以下命令来安装torch-sparse库:

pip install torch_sparse-0.6.10-cp38-cp38-win_amd64.whl  # 根据你的whl文件名命名
  1. 验证是否成功安装了torch-sparse库。可以在Python中运行以下代码来进行验证:
import torch_sparse

如果没有出现错误信息,则说明torch-sparse库已经成功安装并可以正常使用了。

如果你的CUDA版本不是102,你可以在 https://pytorch-geometric.com/whl/torch-1.8.0+cuXXX.html (XXX为你的CUDA版本号) 中找到对应的whl文件进行安装。

希望以上解决方案能帮到你,如果有任何进一步的问题,请随时提问。


如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^