## 查看自己的torch和cuda版本
到自己要安装的虚拟环境中输入以下两个指令
```python
python
print(torch.__version__)
print(torch.version.cuda)
```
## 安装相关依赖
点击进入下面网站,[https://pytorch-geometric.com/whl/](https://pytorch-geometric.com/whl/)
找到自己对应的cuda和torch版本
然后点击进去
https://pytorch-geometric.com/whl/torch-2.5.1%2Bcu118.html
复制网址备用,用于在线安装
输入以下命令
```bath
pip install torch_scatter -f https://pytorch-geometric.com/whl/torch-2.5.1%2Bcu118.html
pip install torch_sparse -f https://pytorch-geometric.com/whl/torch-2.5.1%2Bcu118.html
pip install torch_cluster -f https://pytorch-geometric.com/whl/torch-2.5.1%2Bcu118.html
pip install torch_cluster -f https://pytorch-geometric.com/whl/torch-2.5.1%2Bcu118.html
```
-f后面是刚刚复制的网址
最后再安装PyG
```bath
pip install torch-geometric
```
安装完成以后进入python输入以下两行代码测试一下
```python
>>> import torch_geometric.transforms as Tsform
>>> from torch_geometric.nn import GCNConv,ChebConv
```
没报错则完成安装
PyG安装
最新推荐文章于 2025-04-18 17:00:32 发布