1.安装pytorch1.8.0+cuda11.1

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

2…离线安装包下载：
https://pytorch-geometric.com/whl/
3…离线安装：
最好在从conda虚拟环境下安装。
按照在线安装的顺序，安装4个库：

pip install torch_scatter-2.0.6-cp38-cp38-linux_x86_64.whl
pip install torch_sparse-0.6.9-cp38-cp38-linux_x86_64.whl
pip install torch_cluster-1.5.9-cp38-cp38-linux_x86_64.whl
pip install torch_spline_conv-1.2.1-cp38-cp38-linux_x86_64.whl
pip install torch-geometric

报错：

OSError: libcusparse.so.11: cannot open shared object file: No such file or directory

解决方法：
在.bashrc文件中，最后加上想要加的路径

export PATH=/usr/local/cuda-11.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH

再执行以下

source ~/.bashrc