torch_geometric安装

因为需要跑一些别人的GCN的任务，所以就要安装相关的包；

首先要注意cuda版本和pytorch中的对应关系，必须严格对应，比如我的机器上，cuda版本是10.1；

 

 

 pytorch的cuda版本也是10.1，之前其实是10.2，又重装了一遍；

 

 

pytorch版本是1.4； python版本是3.6；然后在这个网页上

https://pytorch-geometric.com/whl/torch-1.4.0.html

CUDA=cu101
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-spline-conv==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html
pip install torch-cluster==1.4.5
pip install torch-geometric

可以先把各个包按照要求都下下来，然后直接pip install XX就很快了；

这里有个问题需要注意一下，安装torch_geometric，其实默认会安装最新版本（目前是1.6.1），但有些方法在1.4.3之后就不支持了。

比如torch_geometric.utils.scatter_这个方法，在1.5及以后的版本就不支持了，所以代码中有用到这个方法的，要不就另写一个函数代替，要不就把版本退到1.4.3