安装torch_geometric

最新推荐文章于 2025-03-04 01:21:03 发布

CWQF-T

最新推荐文章于 2025-03-04 01:21:03 发布

阅读量2.2k

点赞数 4

文章标签： python pytorch 开发语言

本文链接：https://blog.csdn.net/qq_46080349/article/details/129737928

版权

文章介绍了在PyCharm中安装torch_geometric库的详细步骤，包括寻找匹配的CUDA和PyTorch版本，下载四个必要文件到Anaconda的site-packages目录，然后在虚拟环境中使用pip安装这些文件及torch_geometric本身。最后，通过测试导入torch_geometric模块来确认安装成功。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

torch_geometric是用pycharm学习图神经网络时候会用到的一个包，安装方法如下：

1.先找到对应的包

先在 https://pytorch-geometric.com/whl/ 里面找到对应自己电脑上pytorch 和cuda版本的链接，我的pytorch是1.12，cuda是12.0，但是我在安装pytorch时候选的是cuda版本为11.6所对应的pytorch，所以我应该找的是对应下面第二张图所示的这个链接：torch为1.12，cuda为11.6。

2.下载四个文件

点开上图所示的那个链接之后，再下载对应自己pytorch和cuda版本的下图所示的这四个文件到自己 存放Anaconda3\Lib\site-packages的地址（我的是G:\Software\Anaconda3\Lib\site-packages） 注意：不要放在C盘，否则C盘会爆满！！！

我的anaconda里面python版本是3.9，所以选择cp39的

3.部署这四个文件

在anaconda prompt3里打开自己想要安装torch_geometric的虚拟环境（比如我想在虚拟环境xunipt里面安装）

再按如下步骤转到存放Anaconda3\Lib\site-packages的地址（也就是刚刚放那四个文件的地址，我的是G:\Software\Anaconda3\Lib\site-packages）

然后开始按下面所示顺序pip上述四个文件

pip install torch_scatter-2.0.9-cp39-cp39-win_amd64.whl
pip install torch_sparse-0.6.14-cp39-cp39-win_amd64.whlmd64.whl
pip install torch_cluster-1.6.0+pt112cu116-cp39-cp39-win_amd64.whl
pip install torch_spline_conv-1.2.1+pt112cu116-cp39-cp39-win_amd64.whl
注意：在输入语句时，要把install后面的文件名换成刚刚自己下载好文件的名字

4.安装torch-geometric

成功安装上述四个文件之后再直接输入pip install torch-geometric并且运行，等着它安装好

5.测试torch-geometric是否安装成功

最后可以按如下所示步骤测试一下它是否成功安装

打开anaconda里面的python，输入 import torch_geometric

没有报错，说明安装成功！