图神经网络DGL库安装教程

在之前的文章中介绍了图神经网络的常用库:torch_geometric库的安装教程

本文介绍下图神经网络的另一个常用库DGL的安装教程。DGL官网:Welcome to Deep Graph Library Tutorials and Documentation — DGL 2.0.0 documentation

1.激活虚拟环境

激活需要安装DGl库的虚拟环境,(没有自己的虚拟环境的可以参考我的这篇文章中的(2.创建虚拟环境))。我在本文中演示的虚拟环境名称是Pytorch。

激活环境

打开Anaconda Prompt,开头显示(base)符号,输入指令:

conda activate Pytorch

来激活虚拟环境。

2.查看虚拟环境中的cuda版本和python版本

输入指令:conda list

就会显示当前环境中安装的一些库,我们需要查看cuda和python版本:

我的cuda版本是10.1,python版本是3.7。

注:如果这一步中输入conda list没有显示cudatoolkit,那是因为还没有安装cuda。安装教程可以参看我的文章:MFBZS:Windows配置深度学习环境(虚拟环境中安装cuda+cudnn+pytorch+必要Module)

3.下载dgl文件

去以下网址:

清华源镜像cuda10.1​pypi.tuna.tsinghua.edu.cn/simple/dgl-cu101/

下载cuda10.1+python3.7对应的.whl文件。

其他版本cuda的链接:

清华源镜像cuda10.2​pypi.tuna.tsinghua.edu.cn/simple/dgl-cu102/

清华源镜像cuda10.0​pypi.tuna.tsinghua.edu.cn/simple/dgl-cu100/

清华源镜像cuda9.2​pypi.tuna.tsinghua.edu.cn/simple/dgl-cu92/

清华源镜像cuda9.0​pypi.tuna.tsinghua.edu.cn/simple/dgl-cu90/

这里我选择的是0.5.3版本。

将.whl文件保存到任一目录下:

我的保存目录是:H:\app_app\dgl_py37_py39

4.安装dgl文件

在Anaconda Prompt下,进入到我的保存目录中:依旧是在激活虚拟环境的前提下

然后输入指令:

pip install dgl_cu101-0.5.3-cp37-cp37m-win_amd64.whl

等待安装完成即可。

5.DGL后端修改

初次执行代码会出现以下结果:

可以到以下目录中:

打开config.json文件,可以在"pytorch"处修改三种框架:pytorch 或 mxnet 或 tensorflow

常用的就是pytorch 和 tensorflow。修改后保存即可。

撰写不易,喜欢的话,赞同收藏点赞支持下吧!

关注我,持续更新在跑深度学习过程中遇到的问题!

  • 18
    点赞
  • 26
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值