目录
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