Ubuntu安装torch_geometric

首先，查看自己的cuda版本和pytorch版本：

cuda版本：

cat /usr/local/cuda/version.txt

pytorch版本：

import torch
print(torch.__version__)  

若使用anaconda,也可以直接使用conda list看对应虚拟环境的pytorch版本：

source activate （所建立虚拟环境的名字)  #激活虚拟环境

conda list

知道了这两个版本信息，就可以对照版本获取torch_geometric，比如我的cuda版本是10.2，pytorch版本是1.7.1，那么我会依次输入以下几行命令，大家将自己的cuda版本和pytorch版本替换即可：

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.1+cu102.html

pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.1+cu102.html

pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.7.1+cu102.html

pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.7.1+cu102.html

pip install torch-geometric