之前拷项目遇到了题目中的问题,始终找不到解决方案,后来多方调试、猜测才发现,图中这几个库缺一不可。
之前我没有安装torch-scatter,因为不知道要这样做,根据报错信息以及网上查询也没有找到解决方案,抱着试一试的心理安装了这个库,奇迹般的成功了,因此记录一下。
安装这些库网上有很多帖子,根据torch和cuda版本安装对应的whl文件,再cd到路径下pip install即可。
【以下回答由 GPT 生成】
对于SparseTensor需要安装torch-sparse库的问题,你可以按照以下步骤解决:
import torch
import torch_scipy
如果没有出现错误信息,则说明这两个库都已经正确安装。
nvcc --version
找到适合你CUDA版本的torch-sparse库的whl文件。你可以在以下网址找到对应的whl文件: https://pytorch-geometric.com/whl/torch-1.8.0+cu102.html (假设你的CUDA版本是102)。
下载符合你CUDA版本的torch-sparse库的whl文件,并通过cd命令进入该文件所在路径。
在终端中输入以下命令来安装torch-sparse库:
pip install torch_sparse-0.6.10-cp38-cp38-win_amd64.whl # 根据你的whl文件名命名
import torch_sparse
如果没有出现错误信息,则说明torch-sparse库已经成功安装并可以正常使用了。
如果你的CUDA版本不是102,你可以在 https://pytorch-geometric.com/whl/torch-1.8.0+cuXXX.html (XXX为你的CUDA版本号) 中找到对应的whl文件进行安装。
希望以上解决方案能帮到你,如果有任何进一步的问题,请随时提问。