• DGL安装教程


    诸神缄默不语-个人CSDN博文目录

    DGL官方安装教程网址:Deep Graph Library

    以下仅考虑Linux系统的情况。(在Windows上跑GNN是不是太身残志坚了)

    以下对应可行的安装时间会对应附上。

    注意:这里对应的cuda版本,一般情况下只要跟cudatoolkit对应上就行。

    1. 安装

    2022.8.3
    我安装PyTorch用的Python3.8,命令是conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=10.2 -c pytorch

    可以直接在DGL官网选择对应的情况,复制命令行进行安装:
    在这里插入图片描述

    注意在能够正常使用之前还需要安装psutil包,否则就会报如下错误:

    Traceback (most recent call last):
      File "hello.py", line 1, in 
        import dgl.data
      File "my_env/envgraph1/lib/python3.8/site-packages/dgl/__init__.py", line 25, in 
        from . import dataloading
      File "my_env/lib/python3.8/site-packages/dgl/dataloading/__init__.py", line 10, in 
        from .dataloader import *
      File "my_env/lib/python3.8/site-packages/dgl/dataloading/dataloader.py", line 12, in 
        import psutil
    ModuleNotFoundError: No module named 'psutil'
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10

    安装psutil(psutil的官方下载教程网站:psutil/INSTALL.rst at master · giampaolo/psutil):
    pip install psutil
    在这里插入图片描述

    2. 检验安装效果

    import dgl.data
    
    dataset = dgl.data.CoraGraphDataset()
    print('Number of categories:', dataset.num_classes)
    
    • 1
    • 2
    • 3
    • 4

    输出:

    my_env/lib/python3.8/site-packages/scipy/__init__.py:146: UserWarning: A NumPy version >=1.16.5 and <1.23.0 is required for this version of SciPy (detected version 1.23.1
      warnings.warn(f"A NumPy version >={np_minversion} and <{np_maxversion}"
    Downloading user_path/.dgl/cora_v2.zip from https://data.dgl.ai/dataset/cora_v2.zip...
    Extracting file to user_path/.dgl/cora_v2
    Finished data loading and preprocessing.
      NumNodes: 2708
      NumEdges: 10556
      NumFeats: 1433
      NumClasses: 7
      NumTrainingSamples: 140
      NumValidationSamples: 500
      NumTestSamples: 1000
    Done saving data into cached files.
    Number of categories: 7
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
    • 14

    (这个SciPy和numpy的版本警告我也没辙)

  • 相关阅读:
    Linux 是如何进行内存分配的
    C++设计模式_09_Abstract Factory 抽象工厂
    MySQL数据库索引和事务详解
    grpc 快速入门
    C语言详解(预编译)
    零基础学Java第四节(字符串相关类)
    大数据库题目集——判断题
    导航【C++】
    第八章 通过 REST 使用 Web 会话(Sessions)
    Spring扩展接口(1):ApplicationContextInitializer
  • 原文地址:https://blog.csdn.net/PolarisRisingWar/article/details/126151453