- 知道自己下载的torch配置
cpp
import torch
print(torch.__version__)
print(torch.version.cuda)
- 进入网站,选择自己配置
https://pytorch-geometric.com/whl/ - 下载相应的包
- 安装
cpp
pip install ******.whl