ModuleNotFoundError: No module named 'torch_scatter.cuda'

见：ModuleNotFoundError: No module named ‘torch_sparse.unique_cuda’

原因：没有安装 CUDA,CUDANN， 或者没有配好环境变量。

虽然用

conda install cudatoolkit

可以让 pytorch 使用 gpu 版本，但是系统环境里是没有安装 CUDA 的链接库的，导致使用 pip 安装 torch_sparse 的时候，源码编译时没有找到系统里的 CUDA 库的路径，就默认安装 cpu 版本的torch_sparse，而省略 cuda 相关的模块。