torch-geometric本地whl安装

** torch-geometric(pyg)安装**

一、介绍

最近跑实验需要用到torch-geometric,但是安装老是报错,历经一番艰辛终于搞定了,记录一下。torch-geometric本地whl安装_第1张图片

二、一般安装步骤

1、在安装前要检查电脑的上的torch和cuda版本
(1)import torch; print(torch.version)检查torch版本;
(2)import torch; print(torch.version.cuda)检查cuda版本;
2、检查完后,可以按照torch-geometric官网上的步骤直接用pip或者conda的命令进行安装;
torch-geometric官网链接

三、本地whl安装

因为老是各种报错,所以选择了本地安装。
torch-geometric本地whl安装_第2张图片
本地安装步骤
1、根据官网链接找到自己安装的torch和cuda对应的版本,我自己的是torch1.11+cuda11.3
torch-geometric本地whl安装_第3张图片
2、点进自己安装对应版本的链接,下载torch-geometric的四个相关依赖包,我的Python是3.7,直接下载了最新版本的依赖包
torch-geometric本地whl安装_第4张图片
3、找到下载whl的目录,切换CMD工作路径为.whl文件所在路径:cd+路径
torch-geometric本地whl安装_第5张图片切换路径
4、pip安装.whl文件:

pip install torch_cluster-1.6.0-cp37-cp37m-win_amd64.whl
pip install torch_scatter-2.0.9-cp37-cp37m-win_amd64.whl
pip install torch_sparse-0.6.13-cp37-cp37m-win_amd64.whl
pip install torch_spline_conv-1.2.1-cp37-cp37m-win_amd64.whl

torch-geometric本地whl安装_第6张图片
5、下载torch-geometric,用的豆瓣源:

pip install torch-geometric -i https://pypi.doubanio.com/simple

torch-geometric本地whl安装_第7张图片
换了好几个源都报错,然后找了很多解决方案,最终发现把网络代理关掉,再输入原命令就OK了
torch-geometric本地whl安装_第8张图片
torch-geometric本地whl安装_第9张图片
6、输入下面命令检查torch-geometric安装是否成功,没有报错说明安装导入包torch_geometric成功

python -c "import torch_geometric"

检验成功

小结

主要是想记录总结自己的安装经验,第一次写难免有些不周全的地方,如果有什么问题或者有更方便的安装经验,欢迎大家一起讨论交流。

参考博客

1、手把手教你安装torch_geometric库(pyg)
2、torch-geometric安装教程及常见源下载报错问题
3、【深度学习6】图神经网络库 PyTorch Geometric(PYG)手动安装流程(绝对成功)
4、pip安装本地指定版本的whl文件
5、WARNING: Retrying (Retry(total=4, connect=None, read=None, redirect=None, status=None))解决办法汇总(超全)

你可能感兴趣的:(安装经验,深度学习,pytorch,人工智能,python)