官网: https://pytorch-geometric.readthedocs.io/en/latest/
conda create -n pyG python=3.8
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
#验证torch是否安装成功
print(torch.cuda.is_available())
下载pyG各种依赖包(离线安装不会出现各种问题)
下载地址:
https://data.pyg.org/whl/torch-1.9.0+cu111.html
注意对应cuda和torch版本
安装依赖
例如:torch_cluster-1.5.9-cp38-cp38-linux_x86_64.whl
pip install torch_cluster-1.5.9-cp38-cp38-linux_x86_64.whl 即可(当前路径)
pip install torch-scatter
pip install torch-sparse
pip install torch-geometric
pip install torch-cluster
pip install torch-spline-conv
按顺序安装即可
下载路径:
torch_cluster-1.5.9-cp38-cp38-linux_x86_64.whl
torch_scatter-2.0.9-cp38-cp38-linux_x86_64.whl
torch_sparse-0.6.12-cp38-cp38-linux_x86_64.whl
torch_spline_conv-1.2.1-cp38-cp38-linux_x86_64.whl