Pytorch中安装 torch_geometric 详细图文操作（全）

待我称王封你为后i 2023-01-24 04:30 661阅读 0赞

### 目录 ###

*  前言
 *  1. 安装pytorch
 *  2. 安装torch\_geometric
 *  3. 安装依赖包

# 前言 #

如果已经安装了pytorch版本，对应跳过步骤一，直接安装步骤二即可

注意：依赖包的配置需要根据pytorch版本号以及cuda的版本号，版本号要适配，否则部分函数无法使用

![在这里插入图片描述][284a3716a50f4ed495ad018252538132.png]

# 1. 安装pytorch #

基于代码安装适配的pytorch版本

通过官网进行安装：[pytorch下载官网][pytorch]，此处版本根据代码需要，具体格式如下

*  命令如下（linux）：`conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=10.2 -c pytorch`

![在这里插入图片描述][51124f929dec4397ac33bdc78f4ff78b.png]  
其他操作系统以及版本号需要查看下官网，具体安装格式如上

# 2. 安装torch\_geometric #

在执行命令的时候发现缺失这个`torch_geometric`包

![在这里插入图片描述][40a62bd6d84f4618bf1e88d597dae76e.png]

通过pip install 安装对应的包：

![在这里插入图片描述][3a5b8187d52e4a9daa917fdd565d8bf3.png]

再次执行该代码，发现还是会报错：`ModuleNotFoundError: No module named 'torch_sparse'`

![在这里插入图片描述][3ff695b09d0c43a78a1eac051e71067c.png]

对应还需要如下四个依赖包配置（必须版本适配，否则会出现部分函数无法使用）：

![在这里插入图片描述][c1b77008f05b443d832d64295d035db9.png]

![在这里插入图片描述][108a9986e2e5467683b8a78c6587ca12.png]

# 3. 安装依赖包 #

本身我是装在虚拟环境下，通过`conda list`也可查看到我的配置：

![在这里插入图片描述][2882f3fcb7ef46cf812d857d42a70e0f.png]

通过 [pytorch-geometric官网链接][pytorch-geometric] 下载额外的四个依赖包结合pytorch版本）：

命令如下（无先后顺序）：  
**（注意：不同pytorch以及cud版本不一样，选择的依赖包不一样，通过上面的官网查看，只需替代链接即可）**

*  下载cluster：`wget https://data.pyg.org/whl/torch-1.8.0%2Bcu102/torch_cluster-1.5.9-cp38-cp38-linux_x86_64.whl`
 *  下载scatter：`wget https://data.pyg.org/whl/torch-1.8.0%2Bcu102/torch_scatter-2.0.8-cp38-cp38-linux_x86_64.whl`
 *  下载sparse：`wget https://data.pyg.org/whl/torch-1.8.0%2Bcu102/torch_sparse-0.6.9-cp38-cp38-linux_x86_64.whl`
 *  下载spline：`wget https://data.pyg.org/whl/torch-1.8.0%2Bcu102/torch_spline_conv-1.2.1-cp38-cp38-linux_x86_64.whl`

下载之后，截图如下：

![在这里插入图片描述][8f854cffb5ce40899a61ad45d9ece8fe.png]

对应安装上面四个压缩包：`pip install torch_*`

![在这里插入图片描述][ebedd29eaf9045398d995e7cf1cea1a5.png]  
安装完成看到success即已成功安装该包

[284a3716a50f4ed495ad018252538132.png]: /images/20221119/5fa2474db92949e2915860fb531e90cf.png
[pytorch]: https://pytorch.org/get-started/previous-versions/
[51124f929dec4397ac33bdc78f4ff78b.png]: /images/20221119/169bbe06da3843689baa28c6b957224b.png
[40a62bd6d84f4618bf1e88d597dae76e.png]: /images/20221119/fc84413fc66c40639bcb28024aa66d0d.png
[3a5b8187d52e4a9daa917fdd565d8bf3.png]: /images/20221119/bc7a7695cff5401e9b164db3c9562d39.png
[3ff695b09d0c43a78a1eac051e71067c.png]: /images/20221119/a0f34ffa23084e7ca84f9f6ccd056e2d.png
[c1b77008f05b443d832d64295d035db9.png]: /images/20221119/2a3c61ebd2b145ca8414a680a9d63234.png
[108a9986e2e5467683b8a78c6587ca12.png]: /images/20221119/6afba22b49d2485588cc93abc89567ef.png
[2882f3fcb7ef46cf812d857d42a70e0f.png]: /images/20221119/26b47f302b1d4ae5aebe37488e69ba15.png
[pytorch-geometric]: https://pytorch-geometric.com/whl/
[8f854cffb5ce40899a61ad45d9ece8fe.png]: /images/20221119/a274e013a4594ee8876819e79af79a77.png
[ebedd29eaf9045398d995e7cf1cea1a5.png]: /images/20221119/b0f18ffa9b8d447b9b04a16e7a02448d.png