在win 10 环境下配置torch-scatter,经历了很多坑,最后总结起来,就是要看官方文档,先给出官方文档的连接及相关有用的链接:
https://github.com/rusty1s/pytorch_geometric
https://github.com/pytorch/pytorch
在win 10环境下,cuda的版本只能如下所示:
因此需要安装pytorch 1.4.0版本,cuda 为10.1版本的,其他的版本不支持。
接下来输入如下的命令安装:
pip install torch-scatter==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html
pip install torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html
pip install torch-cluster==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html
pip install torch-spline-conv==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html
pip install torch-geometric
将${CUDA}换为cu101,可成功安装