PyTorch Geometric (PyG) 库的安装

PyG 库的安装

  • 前言
  • 1、PyG库的安装过程
  • 2、版本问题解决

前言

最近在学习图神经网络,需要用到PyG库,发现这个库的安装不能简单的使用pip install 安装,这里记录一下。

1、PyG库的安装过程

第一步:查看自己的torch版本和cuda版本

import torch
print(torch.__version__)
print(torch.version.cuda)
>>2.0.0+cu117

以本人的pytorch为例,pytorch版本是2.0.0,cuda版本是11.7。

第二步:安装相关依赖
点击进入下面网站,https://pytorch-geometric.com/whl/。
点开网站,找到自己的torch和cuda对应的依赖,本人的torch版本是2.0.0,cuda版本是cu117,如下所示:
PyTorch Geometric (PyG) 库的安装_第1张图片
找到对应版本并点击进入,显示如下,复制网址备用
PyTorch Geometric (PyG) 库的安装_第2张图片
接着打开Anaconda Prompt,conda activate pytorch进入到自己的pytorch环境中,在pytorch环境中执行以下命令:

pip install torch_scatter -f https://pytorch-geometric.com/whl/torch-2.0.0%2Bcu117.html
pip install torch_sparse -f https://pytorch-geometric.com/whl/torch-2.0.0%2Bcu117.html
pip install torch_cluster -f https://pytorch-geometric.com/whl/torch-2.0.0%2Bcu117.html
pip install torch_spline_conv -f https://pytorch-geometric.com/whl/torch-2.0.0%2Bcu117.html

注意,-f后面跟的是之前我们复制的网址,即上一个图中画着红线的网址,这个网址是和自己的pytorch版本和cuda版本对应的,每个人的都不一样。
最后执行:

pip install torch-geometric

至此,安装过程结束。

2、版本问题解决

如果之前版本安装错了,可以卸载重新安装,卸载命令是:

pip uninstall torch_geometric torch_scatter torch_sparse torch_cluster torch_spline_conv

你可能感兴趣的:(Pytorch,pytorch,深度学习,python)