默认已经安装好了pytorch和cuda环境
1、检查pytorch和cuda版本
import torch print(torch.__version__) #pytorch版本 print(torch.version.cuda) #cuda版本
2、以pytorch1.12.*及cuda11.6版本为例下载依赖以及包,其他参照官网
pip install torch-scatter -f https://data.pyg.org/whl/torch-1.12.0+cu116.html pip install torch-sparse -f https://data.pyg.org/whl/torch-1.12.0+cu116.html pip install torch-cluster -f https://data.pyg.org/whl/torch-1.12.0+cu116.html pip install torch-spline-conv -f https://data.pyg.org/whl/torch-1.12.0+cu116.html pip install torch-geometric
3、验证(执行以下不报错则安装成功)
import torch import torch_geometric
标签:1.12,https,torch,cuda,install,pip,安装,pyG From: https://www.cnblogs.com/qxybk/p/16882861.html