Pytorch-PyG图神经网络依赖环境安装(Anaconda)

1.默认用户在Anaconda的虚拟环境中已安装Pytorch
2.打开anaconda prompt命令窗, activate "你的虚拟环境名称"
3.在激活后的虚拟环境下输入 python -c "import torch; print(torch.version.cuda)"查看cuda版本,
4.在3后继续输入python -c "import torch; print(torch.version)"查看pytorch版本
5.在下列URL的格式化{}中替换为你的torch和cuda版本:
https://pytorch-geometric.com/whl/torch-{}+{}.html
举个栗子,我的torch版本为1.8.0,cuda版本为10.2,因此我的为https://pytorch-geometric.com/whl/torch-1.8.1+cu101.html
6.打开浏览器,复制上面的URL,分别下载cluster、scatter、sparse、spline_conv这四个依赖包,注意这四者的cpx-cpx要匹配相同,windows环境选择win_amd64,linux环境选择linux_x86_64.
7.继续回到anaconda promt,选择pip install {你在6中下载的4个依赖包的文件路径.whl},很快就能完成。
8.成功完成7后,继续在promt中输入pip install torch-geometric,等待下载完成
9.在promt输入python,并且import torch_geometric,无报错则表明你安装成功,亲测可靠!

为什么这里不选择 pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-{}+{}.html
的方法?这种方法可能极其依赖于网速,因此选择上面的方法会更为可靠些!

DGL依赖库的安装可以参考Blog:https://blog.csdn.net/CY19980216/article/details/110629996

posted @   信海  阅读(214)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 10年+ .NET Coder 心语 ── 封装的思维:从隐藏、稳定开始理解其本质意义
· 提示词工程——AI应用必不可少的技术
· 地球OL攻略 —— 某应届生求职总结
· 字符编码:从基础到乱码解决
· SpringCloud带你走进微服务的世界
点击右上角即可分享
微信分享提示