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