安装PYG

目录

1.通过Anaconda安装

2.通过pip安装

3.尝试历程:

参考


1.通过Anaconda安装

conda install pyg -c pyg -c conda-forge

2.通过pip安装

首先通过如下命令获取系统torch和CUDA版本:

python -c "import torch; print(torch.__version__)"  
python -c "import torch; print(torch.version.cuda)"

然后再将得到的版本号填入下面的${}中 

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric

3.尝试历程:

  • 使用pyg官网co