先前工作:
首先检查查电脑的上的torch和cuda版本
相应代码如下:
import torch;
print(torch.__version__)
print(torch.version.cuda)
安装顺序为:
1.torch-scatter 2.torch-sparse 3.torch-cluster 4.torch-spline-conv 5.torch-geometric
相应包可下载至本地(https://data.pyg.org/whl/)
注意:python版本、pyorch版本、CUDA版本都要相对应
最后安装geometric(关闭外网)
pip install torch-geometric -i https://pypi.tuna.tsinghua.edu.cn/simple
测试
导入未报错说明已经安装成功