参照Installation — pytorch_geometric 1.6.1 documentation
1、确保安装 PyTorch版本:
python -c "import torch; print(torch.__version__)"
1.6.0+cu101
2、查找安装 PyTorch 的 CUDA 版本:
python -c "import torch; print(torch.version.cuda)"
10.1
3、安装相关包:
pip install torch-scatter==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-sparse==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-cluster==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-spline-conv==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-geometric