ModuleNotFoundError: No module named ‘torch_sparse‘

https://github.com/rusty1s/pytorch_sparse

1、
You can now install pytorch-sparse via Anaconda for all major OS/PyTorch/CUDA combinations 🤗 Given that you have pytorch >= 1.8.0 installed, simply run

conda install pytorch-sparse -c pyg

2、
PyTorch 1.10.0
To install the binaries for PyTorch 1.10.0, simply run

pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-1.10.0+${CUDA}.html

where ${CUDA} should be replaced by either cpu, cu102, or cu113 depending on your PyTorch installation.

我用的第二种方法,我的cuda是11.5,命令如下:

pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-1.10.0+cu113.html

查看pytorch版本:

import torch
print(torch.__version__)  #注意是双下划线

版权声明:本文为CSDN博主「光子乘羽」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/hanshuning/article/details/121682424

光子乘羽

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

实战 | 实时的目标检测与识别简单应用

吃粽子迎端午计算机视觉研究院专栏作者:Edison_G最近总是有很多入门的朋友问我,我进入计算机视觉这个领域难不难?是不是要学习很多知识?到底哪个方向比较好?长按扫描二维码关

小目标检测总结

小目标检测总结
小目标的定义:图像中极少的小目标(32像素×32像素一下)
注:一般来说下采样率为2,一般的特征提取网络通常进行五次下采样。而五次下采样的倍率为2的五次方