ModuleNotFoundError: No module named ‘torch_sparse‘

ModuleNotFoundError: No module named 'torch_sparse'

原因：

pip install torch_sparse 时，编译器没有找到 cuda 或 cudann 的环境变量，只生成了 cpu 版本的 torch_sparse，而省略了和cuda相关的模块。

解决方案：

手动安装 cuda，cudann

参考教程：

https://blog.youkuaiyun.com/Mind_programmonkey/article/details/99688839#commentBox

cudnn的安装

（1）下载安装文件

按需求下载cudnn的安装文件：https://developer.nvidia.com/rdp/cudnn-archive
需要注册账号

（2）安装cudnn

解压下载的文件，可以看到cuda文件夹，在当前目录打开终端，执行如下命令：

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/

sudo chmod a+r /usr/local/cuda/include/cudnn.h

sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
（3）查看cudnn版本

在终端输入

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果出现下图所示版本信息，说明安装成功。