torch_geometric是用pycharm学习图神经网络时候会用到的一个包,安装方法如下:
先在 https://pytorch-geometric.com/whl/ 里面找到对应自己电脑上pytorch 和cuda版本的链接,我的pytorch是1.12,cuda是12.0,但是我在安装pytorch时候选的是cuda版本为11.6所对应的pytorch,所以我应该找的是对应下面第二张图所示的这个链接:torch为1.12,cuda为11.6。
点开上图所示的那个链接之后,再下载对应自己pytorch和cuda版本的下图所示的这四个文件到自己 存放Anaconda3\Lib\site-packages的地址(我的是G:\Software\Anaconda3\Lib\site-packages) 注意:不要放在C盘,否则C盘会爆满!!!
我的anaconda里面python版本是3.9,所以选择cp39的
在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后面的文件名换成刚刚自己下载好文件的名字
成功安装上述四个文件之后再直接输入pip install torch-geometric并且运行,等着它安装好
最后可以按如下所示步骤测试一下它是否成功安装
打开anaconda里面的python,输入 import torch_geometric
没有报错,说明安装成功!