首页 如何安装PyTorch Geometric
文章
取消

如何安装PyTorch Geometric

首先,附上geometric官网链接:geometric官网

如果你不想看geometric官网,那么往下看:

首先进入自己的环境,按顺序输入如下五行(当然需要替换了):

1
2
3
4
5
pip install torch-scatter==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-${TORCH}.html
pip install torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-${TORCH}.html
pip install torch-cluster==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-${TORCH}.html
pip install torch-spline-conv==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-${TORCH}.html
pip install torch-geometric

然后即可安装成功。

即,如果你是CUDA10.2,torch1.5.0,则输入:

1
2
3
4
5
pip install torch-scatter==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.5.0.html
pip install torch-sparse==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.5.0.html
pip install torch-cluster==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.5.0.html
pip install torch-spline-conv==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.5.0.html
pip install torch-geometric

安装成功!

本文由作者按照 CC BY 4.0 进行授权