参考博客pyg安装+pytorch_pyg版本对照-CSDN博客,主要记录学习过程。
本机环境说明:
python==3.8.18
torch==2.1.0
要安装torch_geometric2.2.0
安装步骤:
1.查看本机环境python和pytorch版本,之后在这里https://pytorch-geometric.com/whl/找到与自己版本相符的链接
在环境文件中新建一个pyg文件(这里文件夹在哪应该无所谓,最后在命令行窗口打开就可以即cd +地址),在以下四种文件中选择合适的进行下载:
2.在Anaconda Prompt界面中进入环境,并安装:
输入cd C:\Users\24205\miniconda3\envs\Slm_pytorch\pyg 进入刚刚下载文件的目录:
运行 pip install torch_geometric 下载,到这一步还没有结束,如果直接import torch_geometric会报错。需要安装另外四个文件。
3.输入pip install +第一步下载好的文件名称,如下(按顺序下载):
pip install torch_scatter-2.1.2+pt21cpu-cp38-cp38-win_amd64.whl
pip install torch_sparse-0.6.18+pt21cpu-cp38-cp38-win_amd64.whl
pip install torch_cluster-1.6.3+pt21cpu-cp38-cp38-win_amd64.whl
pip install torch_spline_conv-1.2.2+pt21cpu-cp38-cp38-win_amd64.whl
4.检查是否安装成功
进入python编译模式
输入:
import torch_geometric
安装成功!!