如何安装PyTorch Geometric

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

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

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

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,则输入:

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

安装成功!

你可能感兴趣的:(如何安装PyTorch Geometric)