PyG安装

## 查看自己的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
```
没报错则完成安装

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值