1. torch_geometric 环境搭建


找到自己对应的版本即可

我的 CUDA版本 : 10.2
Pytorch版本 : 1.10.0

image-20220304170554471

  • 根据具体情况选择版本
image-20220304171143972
pip install torch-scatter==2.0.9 -f https://pytorch-geometric.com/whl/torch-1.10.0+cu102.html
pip install torch-cluster==1.5.9 -f https://pytorch-geometric.com/whl/torch-1.10.0+cu102.html
pip install torch-spline-conv==1.2.1 -f https://pytorch-geometric.com/whl/torch-1.10.0+cu102.html
pip install torch-sparse==0.6.12 -f https://pytorch-geometric.com/whl/torch-1.10.0+cu102.html
pip install torch-geometric
Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐