因为需要跑一些别人的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