见:ModuleNotFoundError: No module named ‘torch_sparse.unique_cuda’
原因:没有安装 CUDA,CUDANN, 或者没有配好环境变量。
虽然用
conda install cudatoolkit
可以让 pytorch 使用 gpu 版本,但是系统环境里是没有安装 CUDA 的链接库的,导致使用 pip 安装 torch_sparse 的时候,源码编译时没有找到系统里的 CUDA 库的路径,就默认安装 cpu 版本的torch_sparse,而省略 cuda 相关的模块。